1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
//@HEADER
// ************************************************************************
//
// Kokkos v. 4.0
// Copyright (2022) National Technology & Engineering
// Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
// See https://kokkos.org/LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER
#include <TestHIP_Category.hpp>
#include <Kokkos_Core.hpp>
namespace Test {
// On MI300a with ROCM <= 6.2.1, hipMemsetAsync was failing with an error when
// called on host-allocated buffers. The fix was in PR 7380 to use a
// parallel_for to zero memory
template <typename MemorySpace>
void unified_memory_zero_memset() {
static_assert(Kokkos::is_memory_space_v<MemorySpace>);
constexpr size_t N = static_cast<size_t>(1024 * 1024); // size doesn't matter
std::vector<int> v(N, 1); // initialize to non-zero
Kokkos::View<int*, MemorySpace> a(v.data(), N);
// zero with deep_copy (this is where the error occured)
Kokkos::deep_copy(a, 0);
// see if it was zeroed
int err;
Kokkos::parallel_reduce(
N, KOKKOS_LAMBDA(int i, int& lerr) { lerr += (a[i] != 0); }, err);
EXPECT_EQ(err, 0);
}
TEST(hip, unified_memory_zero_memset) {
#if !defined(KOKKOS_IMPL_HIP_UNIFIED_MEMORY)
GTEST_SKIP()
<< "this test should only be run with HIP unified memory enabled";
#endif
unified_memory_zero_memset<Kokkos::SharedSpace>();
unified_memory_zero_memset<Kokkos::HIPSpace>();
}
} // namespace Test
|
