File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments