Skip to content

Commit a193049

Browse files
authored
add parallel for to execute MOSAIC module data on GPU (#104)
1 parent cda3d97 commit a193049

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/unit-tests/TChem_Test_MOSAIC.hpp

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,25 @@ void static MOSAIC_Test() {
1515
using real_type_2d_view = TChem::real_type_2d_view;
1616
using real_type_2d_view_host = TChem::real_type_2d_view_host;
1717

18+
std::string profile_name ="Verification_test_MTEM_compute_log_gamZ";
19+
using policy_type =
20+
typename TChem::UseThisTeamPolicy<TChem::exec_space>::type;
1821
const auto exec_space_instance = TChem::exec_space();
22+
const auto host_exec_space = TChem::host_exec_space();
23+
policy_type policy(exec_space_instance, 1, Kokkos::AUTO());
1924

2025
const auto mmd = TChem::Impl::MosaicModelData<device_type>();
2126

27+
// Check this routines on GPUs.
28+
Kokkos::parallel_for(
29+
profile_name,
30+
policy,
31+
KOKKOS_LAMBDA(const typename policy_type::member_type& member) {
32+
2233
auto b_mtem = mmd.b_mtem.template view<device_type>();
2334

2435
EXPECT_TRUE(b_mtem(0,mmd.jnh4so4,mmd.jnh4hso4) == -4.13219);
25-
36+
});
2637
}
2738

2839
}// namespace Test

0 commit comments

Comments
 (0)