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
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#include <TestHIP_Category.hpp>
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
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
|
