[CUDA][Bug] CSR transpose bug in CUDA 12 (#7295)

2ff3006c · Muhammed Fatih BALIN · GitHub · 20e5e266 · 2ff3006c
Unverified Commit 2ff3006c authored Apr 12, 2024 by Muhammed Fatih BALIN Committed by GitHub Apr 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/array/cuda/csr_transpose.cc src/array/cuda/csr_transpose.cc +4 -0

No files found.
--- a/src/array/cuda/csr_transpose.cc
+++ b/src/array/cuda/csr_transpose.cc
@@ -22,6 +22,7 @@ CSRMatrix CSRTranspose(CSRMatrix csr) {

 template <>
 CSRMatrix CSRTranspose<kDGLCUDA, int32_t>(CSRMatrix csr) {
+#if CUDART_VERSION < 12000
  auto* thr_entry = runtime::CUDAThreadEntry::ThreadLocal();
  cudaStream_t stream = runtime::getCurrentCUDAStream();
  // allocate cusparse handle if needed
@@ -76,6 +77,9 @@ CSRMatrix CSRTranspose<kDGLCUDA, int32_t>(CSRMatrix csr) {

  return CSRMatrix(
      csr.num_cols, csr.num_rows, t_indptr, t_indices, t_data, false);
+#else
+  return COOToCSR(COOTranspose(CSRToCOO(csr, false)));
+#endif
 }

 template <>