Fix HIP and CUDA ELL SpMV Debug error

ginkgo-project · Mar 17, 2021 · 281447e · 281447e
1 parent b3ea4bb
commit 281447e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cuda/matrix/ell_kernels.cu b/cuda/matrix/ell_kernels.cu
@@ -151,8 +151,8 @@ void abstract_spmv(syn::value_list<int, info>, int num_worker_per_row,
         std::array<size_type, 1>{{num_stored_elements_per_row * stride}},
         a->get_const_values());
     const auto b_vals = gko::acc::range<b_accessor>(
-        std::array<size_type, 2>{{nrows, b->get_stride()}},
-        b->get_const_values());
+        std::array<size_type, 2>{{b->get_size()[0], b->get_size()[1]}},
+        b->get_const_values(), std::array<size_type, 1>{{b->get_stride()}});
 
     if (alpha == nullptr && beta == nullptr) {
         kernel::spmv<num_thread_per_worker, atomic>

diff --git a/hip/matrix/ell_kernels.hip.cpp b/hip/matrix/ell_kernels.hip.cpp
@@ -155,8 +155,8 @@ void abstract_spmv(syn::value_list<int, info>, int num_worker_per_row,
         std::array<size_type, 1>{{num_stored_elements_per_row * stride}},
         a->get_const_values());
     const auto b_vals = gko::acc::range<b_accessor>(
-        std::array<size_type, 2>{{nrows, b->get_stride()}},
-        b->get_const_values());
+        std::array<size_type, 2>{{b->get_size()[0], b->get_size()[1]}},
+        b->get_const_values(), std::array<size_type, 1>{{b->get_stride()}});
 
     if (alpha == nullptr && beta == nullptr) {
         hipLaunchKernelGGL(