Update sparse_matrix.cpp

taichi-dev · Aug 4, 2022 · e9e4691 · e9e4691
1 parent 13dc40a
commit e9e4691
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taichi/program/sparse_matrix.cpp b/taichi/program/sparse_matrix.cpp
@@ -257,7 +257,7 @@ void CuSparseMatrix::spmv(Program *prog, const Ndarray &x, Ndarray &y) {
 
   void *dBuffer = NULL;
   CUDAContext::get_instance().make_current();
-  CUDADriver::get_instance().malloc(&dBuffer, 8);
+  CUDADriver::get_instance().malloc(&dBuffer, bufferSize);
   printf("spmv bufferSize: %zu\n", bufferSize);
   CUSPARSEDriver::get_instance().cpSpMV(
       cusparse_handle, CUSPARSE_OPERATION_NON_TRANSPOSE, &alpha, matrix_, vecX,