From 198a7af13095f14bce93b61806c86e42d80bf330 Mon Sep 17 00:00:00 2001 From: Yuuichi Asahi Date: Tue, 7 Jan 2025 16:38:43 +0900 Subject: [PATCH] remove the global namespace Signed-off-by: Yuuichi Asahi --- .../unit_test/Test_Batched_SerialGemm.hpp | 15 +- .../Test_Batched_SerialGemm_Complex.hpp | 238 ++++++++++-------- .../Test_Batched_SerialGemm_Real.hpp | 100 ++++---- 3 files changed, 189 insertions(+), 164 deletions(-) diff --git a/batched/dense/unit_test/Test_Batched_SerialGemm.hpp b/batched/dense/unit_test/Test_Batched_SerialGemm.hpp index 32f58675b5..adc9a51aac 100644 --- a/batched/dense/unit_test/Test_Batched_SerialGemm.hpp +++ b/batched/dense/unit_test/Test_Batched_SerialGemm.hpp @@ -25,8 +25,6 @@ #include "KokkosKernels_TestUtils.hpp" #include "KokkosKernels_TestVanilla.hpp" -using namespace KokkosBatched; - namespace Test { namespace Gemm { @@ -53,8 +51,9 @@ struct Functor_TestBatchedSerialGemm { auto bb = Kokkos::subview(m_b, k, Kokkos::ALL(), Kokkos::ALL()); auto cc = Kokkos::subview(m_c, k, Kokkos::ALL(), Kokkos::ALL()); - info += SerialGemm::invoke( - m_alpha, aa, bb, m_beta, cc); + info += + KokkosBatched::SerialGemm::invoke( + m_alpha, aa, bb, m_beta, cc); } inline int run() { @@ -107,10 +106,10 @@ void impl_test_batched_gemm(const int N, const int matAdim1, const int matAdim2, Kokkos::deep_copy(C_ref, C); Functor_BatchedVanillaGEMM vgemm; - vgemm.A_t = !std::is_same_v; - vgemm.B_t = !std::is_same_v; - vgemm.A_c = std::is_same_v; - vgemm.B_c = std::is_same_v; + vgemm.A_t = !std::is_same_v; + vgemm.B_t = !std::is_same_v; + vgemm.A_c = std::is_same_v; + vgemm.B_c = std::is_same_v; vgemm.A = A; vgemm.B = B; vgemm.C = C_ref; diff --git a/batched/dense/unit_test/Test_Batched_SerialGemm_Complex.hpp b/batched/dense/unit_test/Test_Batched_SerialGemm_Complex.hpp index 0d751869ad..ab97732238 100644 --- a/batched/dense/unit_test/Test_Batched_SerialGemm_Complex.hpp +++ b/batched/dense/unit_test/Test_Batched_SerialGemm_Complex.hpp @@ -18,105 +18,116 @@ /// fcomplex, fcomplex TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_nt_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_t_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_c_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_c_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_c_fcomplex_fcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, Kokkos::complex, param_tag_type, Algo::Gemm::Blocked>(); + using param_tag_type = + ::Test::Gemm::ParamTag; + test_batched_gemm, Kokkos::complex, param_tag_type, + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } /// fcomplex, float TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_nt_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_t_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_c_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_c_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_c_fcomplex_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, float, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, float, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = + ::Test::Gemm::ParamTag; + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, float, param_tag_type, KokkosBatched::Algo::Gemm::Unblocked>(); } #endif @@ -126,114 +137,125 @@ TEST_F(TestCategory, batched_scalar_serial_gemm_c_c_fcomplex_float) { /// dcomplex, dcomplex TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_nt_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_t_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_c_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_c_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_c_dcomplex_dcomplex) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = + ::Test::Gemm::ParamTag; test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Blocked>(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm, Kokkos::complex, param_tag_type, - Algo::Gemm::Unblocked>(); + KokkosBatched::Algo::Gemm::Unblocked>(); } /// dcomplex, double TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_nt_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_t_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_c_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_c_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_c_c_dcomplex_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm, double, param_tag_type, Algo::Gemm::Blocked>(); - test_batched_gemm, double, param_tag_type, Algo::Gemm::Unblocked>(); + using param_tag_type = + ::Test::Gemm::ParamTag; + test_batched_gemm, double, param_tag_type, KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm, double, param_tag_type, + KokkosBatched::Algo::Gemm::Unblocked>(); } #endif diff --git a/batched/dense/unit_test/Test_Batched_SerialGemm_Real.hpp b/batched/dense/unit_test/Test_Batched_SerialGemm_Real.hpp index f2028775ad..0192b61b0f 100644 --- a/batched/dense/unit_test/Test_Batched_SerialGemm_Real.hpp +++ b/batched/dense/unit_test/Test_Batched_SerialGemm_Real.hpp @@ -15,113 +15,117 @@ //@HEADER #if defined(KOKKOS_BHALF_T_IS_FLOAT) TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_bhalf_bhalf) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_bhalf_bhalf) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_bhalf_bhalf) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Unblocked>(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_bhalf_bhalf) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Unblocked>(); } #endif // KOKKOS_BHALF_T_IS_FLOAT #if defined(KOKKOS_HALF_T_IS_FLOAT) TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_half_half) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_half_half) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_half_half) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_half_half) { - using param_tag_type = ::Test::Gemm::ParamTag; + using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); test_batched_gemm(); + KokkosBatched::Algo::Gemm::Blocked>(); + test_batched_gemm(); } #endif // KOKKOS_HALF_T_IS_FLOAT #if defined(KOKKOSKERNELS_INST_FLOAT) TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_float_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_float_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_float_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_float_float) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } #endif #if defined(KOKKOSKERNELS_INST_DOUBLE) TEST_F(TestCategory, batched_scalar_serial_gemm_nt_nt_double_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_nt_double_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_nt_t_double_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } TEST_F(TestCategory, batched_scalar_serial_gemm_t_t_double_double) { - using param_tag_type = ::Test::Gemm::ParamTag; - test_batched_gemm(); - test_batched_gemm(); + using param_tag_type = ::Test::Gemm::ParamTag; + test_batched_gemm(); + test_batched_gemm(); } #endif