Skip to content

Commit

Permalink
Merge update cuda auto csr param
Browse files Browse the repository at this point in the history
This PR updates the parameters to select the strategy in automatical csr.

Related PR: #407
  • Loading branch information
yhmtsai authored Dec 2, 2019
2 parents 4e2b9aa + 196fd04 commit af3af30
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions include/ginkgo/core/matrix/csr.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -461,12 +461,10 @@ class Csr : public EnableLinOp<Csr<ValueType, IndexType>>,
// if the number of stored elements is larger than <nnz_limit> or
// the maximum number of stored elements per row is larger than
// <row_len_limit>, 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;
Expand Down

0 comments on commit af3af30

Please sign in to comment.