fix: reintroduce RealType in impl_test_batched_iamax

Signed-off-by: Yuuichi Asahi <y.asahi@nr.titech.ac.jp>
kokkos · Oct 28, 2024 · 5b53a2a · 5b53a2a
1 parent 9e7fc77
commit 5b53a2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/batched/dense/unit_test/Test_Batched_SerialIamax.hpp b/batched/dense/unit_test/Test_Batched_SerialIamax.hpp
@@ -162,6 +162,8 @@ void impl_test_batched_iamax_analytical(const std::size_t N) {
 /// \param BlkSize [in] Block size of matrix A
 template <typename DeviceType, typename ScalarType, typename LayoutType>
 void impl_test_batched_iamax(const std::size_t N, const std::size_t BlkSize) {
+  using ats               = typename Kokkos::ArithTraits<ScalarType>;
+  using RealType          = typename ats::mag_type;
   using View2DType        = Kokkos::View<ScalarType **, LayoutType, DeviceType>;
   using StridedView2DType = Kokkos::View<ScalarType **, Kokkos::LayoutStride, DeviceType>;
   using MaxView1DType     = Kokkos::View<int *, LayoutType, DeviceType>;