From 4dd65a6643467e0e6547d354fb7b9e31e0fb1f49 Mon Sep 17 00:00:00 2001
From: Luc Berger-Vergiat <lberge@sandia.gov>
Date: Thu, 21 Sep 2023 14:42:03 -0600
Subject: [PATCH] Sparse: one more change after review

---
 sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp b/sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp
index df07db0aa9..d0e03c3bf8 100644
--- a/sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp
+++ b/sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp
@@ -83,15 +83,7 @@ int64_t crsmatrix_traversal_launch_parameters(int64_t numRows, int64_t nnz,
 
   if (nnz_per_row < 1) nnz_per_row = 1;
 
-  int max_vector_length = 1;
-#ifdef KOKKOS_ENABLE_CUDA
-  if (std::is_same<execution_space, Kokkos::Cuda>::value)
-    max_vector_length = 32;
-#endif
-#ifdef KOKKOS_ENABLE_HIP
-  if (std::is_same<execution_space, Kokkos::Experimental::HIP>::value)
-    max_vector_length = 64;
-#endif
+  int max_vector_length = Kokkos::TeamPolicy<execution_space>::vector_length_max();
 
   if (vector_length < 1) {
     vector_length = 1;