Skip to content

Commit 99e97eb

Browse files
committed
rebasing
1 parent 8208455 commit 99e97eb

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

cuBLAS/sp_gemm.hh

+12
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,18 @@ class sp_gemm_gpu : public sp_gemm<T> {
155155

156156
switch(offload_) {
157157
case gpuOffloadType::always: {
158+
// Make matrix descriptors
159+
cusparseCheckError(
160+
cusparseCreateCsr(&descrA_, n_, n_, A_nnz_, A_row_dev_,
161+
A_col_dev_, A_val_dev_, rType_, cType_,
162+
indType_, cudaDataType_));
163+
cusparseCheckError(
164+
cusparseCreateCsr(&descrB_, n_, n_, B_nnz_, B_row_dev_,
165+
B_col_dev_, B_val_dev_, rType_, cType_,
166+
indType_, cudaDataType_));
167+
cusparseCheckError(
168+
cusparseCreateCsr(&descrC_, n_, n_, 0, C_row_dev_, NULL, NULL,
169+
rType_, cType_, indType_, cudaDataType_));
158170
break;
159171
}
160172
case gpuOffloadType::once: {

0 commit comments

Comments
 (0)