From e87f9914de58a407b9f454c73ab21c93f3adc57a Mon Sep 17 00:00:00 2001 From: Christian Glusa Date: Tue, 19 Nov 2019 15:23:37 -0700 Subject: [PATCH] Fix suggested vector size when nr == 0 The local part of a matrix can have zero rows. --- .../kokkos-kernels/src/common/KokkosKernels_ExecSpaceUtils.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/packages/kokkos-kernels/src/common/KokkosKernels_ExecSpaceUtils.hpp b/packages/kokkos-kernels/src/common/KokkosKernels_ExecSpaceUtils.hpp index c0fbd2fec7f2..17e90d7bd094 100644 --- a/packages/kokkos-kernels/src/common/KokkosKernels_ExecSpaceUtils.hpp +++ b/packages/kokkos-kernels/src/common/KokkosKernels_ExecSpaceUtils.hpp @@ -103,7 +103,8 @@ inline int kk_get_suggested_vector_size( break; case Exec_CUDA: - suggested_vector_size_ = nnz / double (nr) + 0.5; + if (nr > 0) + suggested_vector_size_ = nnz / double (nr) + 0.5; if (suggested_vector_size_ < 3){ suggested_vector_size_ = 2; }