ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v4.1.0

Browse Source
This commit is contained in:
Stan Gerald Moore
2023-06-29 10:42:42 -06:00
parent 170173a213
commit 330107b77b
480 changed files with 24051 additions and 23393 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
lib/kokkos/containers/unit_tests/TestDynViewAPI_rank12345.hpp
View File

@ -20,4 +20,56 @@ namespace Test {
TEST(TEST_CATEGORY, dyn_rank_view_api_operator_rank12345) {
TestDynViewAPI<double, TEST_EXECSPACE>::run_operator_test_rank12345();
}
template <typename SharedMemorySpace>
void test_dyn_rank_view_resize() {
int n = 1000;
Kokkos::DynRankView<double, SharedMemorySpace> device_view("device view", n);
// Make sure we don't deallocate memory in Kokkos::resize
auto device_view_copy = device_view;
Kokkos::resize(device_view, 2 * n);
// Loop in reverse to increase likelihood of missing fence detection assuming
// that resize copies values in order.
for (int i = 2 * n - 1; i >= 0; --i) device_view(i) = i + 1;
Kokkos::fence();
// Check that Kokkos::resize completed before setting the values on the host
// manually (possibly because of missing fences).
for (int i = 0; i < 2 * n; ++i) ASSERT_EQ(device_view(i), i + 1);
}
template <typename SharedMemorySpace>
void test_dyn_rank_view_realloc() {
int n = 1000;
Kokkos::DynRankView<double, SharedMemorySpace> device_view("device view", n);
// Make sure we don't deallocate memory in Kokkos::realloc
auto device_view_copy = device_view;
Kokkos::realloc(device_view, 2 * n);
// Loop in reverse to increase likelihood of missing fence detection assuming
// that realloc sets values in order.
for (int i = 2 * n - 1; i >= 0; --i) device_view(i) = i + 1;
Kokkos::fence();
// Check that Kokkos::realloc completed before setting the values on the host
// manually (possibly because of missing fences).
for (int i = 0; i < 2 * n; ++i) ASSERT_EQ(device_view(i), i + 1);
}
#ifdef KOKKOS_HAS_SHARED_SPACE
TEST(TEST_CATEGORY, dyn_rank_view_check_fence_resize_realloc) {
if constexpr (std::is_same_v<TEST_EXECSPACE, Kokkos::DefaultExecutionSpace>) {
test_dyn_rank_view_resize<Kokkos::SharedSpace>();
test_dyn_rank_view_realloc<Kokkos::SharedSpace>();
} else {
GTEST_SKIP() << "skipping since not default execution space";
}
}
#endif
} // namespace Test