Skip to content

Commit

Permalink
Exclude CUDA 11.3.0 from the triangular solver
Browse files Browse the repository at this point in the history
Only CUDA 11.3.1 starts to support `cusparseSpSM_solve`, version 11.3.0
does not support it yet.
  • Loading branch information
Thomas Grützmacher committed Jan 12, 2022
1 parent 388e91f commit c5e1297
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
8 changes: 4 additions & 4 deletions cuda/base/cusparse_bindings.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -879,7 +879,7 @@ inline void destroy(cusparseSpMatDescr_t descr)
}


#if (CUDA_VERSION >= 11030)
#if (CUDA_VERSION > 11030)


template <typename AttribType>
Expand All @@ -905,7 +905,7 @@ inline void destroy(cusparseSpSMDescr_t info)
}


#endif // CUDA_VERSION >= 11030
#endif // CUDA_VERSION > 11030


#endif // defined(CUDA_VERSION) && (CUDA_VERSION >= 11000)
Expand Down Expand Up @@ -1108,7 +1108,7 @@ GKO_BIND_CUSPARSE64_CSRSM2_SOLVE(ValueType, detail::not_implemented);
#undef GKO_BIND_CUSPARSE64_CSRSM2_SOLVE


#if (defined(CUDA_VERSION) && (CUDA_VERSION >= 11030))
#if (defined(CUDA_VERSION) && (CUDA_VERSION > 11030))


template <typename ValueType>
Expand Down Expand Up @@ -1153,7 +1153,7 @@ void spsm_solve(cusparseHandle_t handle, cusparseOperation_t op_a,
}


#endif // (defined(CUDA_VERSION) && (CUDA_VERSION >= 11030))
#endif // (defined(CUDA_VERSION) && (CUDA_VERSION > 11030))


template <typename IndexType>
Expand Down
2 changes: 1 addition & 1 deletion cuda/solver/common_trs_kernels.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ namespace cuda {
namespace {


#if (defined(CUDA_VERSION) && (CUDA_VERSION >= 11030))
#if (defined(CUDA_VERSION) && (CUDA_VERSION > 11030))


template <typename ValueType, typename IndexType>
Expand Down
2 changes: 1 addition & 1 deletion cuda/test/solver/lower_trs_kernels.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,7 @@ TEST_F(LowerTrs, CudaMultipleRhsApplyIsEquivalentToRef)
d_b2->convert_to(db2_strided.get());
// The cuSPARSE Generic SpSM implementation uses the wrong stride here
// so the input and output stride need to match
#if CUDA_VERSION >= 11030
#if CUDA_VERSION > 11030
auto dx_strided = Mtx::create(cuda, x->get_size(), 4);
#else
auto dx_strided = Mtx::create(cuda, x->get_size(), 5);
Expand Down

0 comments on commit c5e1297

Please sign in to comment.