Fix for gemm

kokkos · Jan 11, 2024 · d55c19c · d55c19c
1 parent 5290aaf
commit d55c19c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sparse/impl/KokkosSparse_spiluk_numeric_impl.hpp b/sparse/impl/KokkosSparse_spiluk_numeric_impl.hpp
@@ -107,6 +107,8 @@ struct IlukWrap {
     WorkViewType iw;
     lno_t lev_start;
 
+    using reftype = scalar_t&;
+
     Common(const ARowMapType &A_row_map_, const AEntriesType &A_entries_,
            const AValuesType &A_values_, const LRowMapType &L_row_map_,
            const LEntriesType &L_entries_, LValuesType &L_values_,
@@ -238,6 +240,8 @@ struct IlukWrap {
         typename AValuesType::device_type,
         Kokkos::MemoryTraits<Kokkos::Unmanaged | Kokkos::RandomAccess> >;
 
+    using reftype = LValuesUnmanaged2DBlockType;
+
     Common(const ARowMapType &A_row_map_, const AEntriesType &A_entries_,
            const AValuesType &A_values_, const LRowMapType &L_row_map_,
            const LEntriesType &L_entries_, LValuesType &L_values_,
@@ -472,7 +476,7 @@ struct IlukWrap {
               const auto col  = Base::U_entries(kk);
               const auto ipos = Base::iw(my_team, col);
               if (ipos != -1) {
-                auto C = col < rowid ? Base::lget(ipos) : Base::uget(ipos);
+                typename Base::reftype C = col < rowid ? Base::lget(ipos) : Base::uget(ipos);
                 Base::gemm(Base::uget(kk), fact, C);
               }
             });  // end for kk