From 196fd04dacf3390193ac0529bd1bffbb8e1d21d2 Mon Sep 17 00:00:00 2001 From: "Yuhsiang M. Tsai" Date: Fri, 29 Nov 2019 14:21:56 +0100 Subject: [PATCH] update cuda auto csr param --- include/ginkgo/core/matrix/csr.hpp | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/include/ginkgo/core/matrix/csr.hpp b/include/ginkgo/core/matrix/csr.hpp index 83bad0ce24e..75e94b800f8 100644 --- a/include/ginkgo/core/matrix/csr.hpp +++ b/include/ginkgo/core/matrix/csr.hpp @@ -461,12 +461,10 @@ class Csr : public EnableLinOp>, // if the number of stored elements is larger than or // the maximum number of stored elements per row is larger than // , use load_balance otherwise use classical - // CUDA: nnz_limit = 1e6, row_len_limit = 64 - // TODO: need to tune CUDA parameters according to new classical - // strategy + // CUDA: nnz_limit = 1e6, row_len_limit = 1024 // AMD: nnz_limit = 1e8, row_len_limit = 768 index_type nnz_limit = 1e6; - index_type row_len_limit = 64; + index_type row_len_limit = 1024; #if GINKGO_HIP_PLATFORM_HCC if (!cuda_strategy_) { nnz_limit = 1e8;