fix: reintroduce RealType in impl_test_batched_iamax

Signed-off-by: Yuuichi Asahi <[email protected]>
kokkos · Oct 27, 2024 · c0ccdb9 · c0ccdb9
1 parent b0bac8e
commit c0ccdb9
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>;