print

taichi-dev · Aug 11, 2022 · 38fa224 · 38fa224
1 parent d851b48
commit 38fa224
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taichi/program/sparse_matrix.cpp b/taichi/program/sparse_matrix.cpp
@@ -254,11 +254,11 @@ void CuSparseMatrix::spmv(Program *prog, const Ndarray &x, Ndarray &y) {
   CUSPARSEDriver::get_instance().cpSpMV_bufferSize(
       cusparse_handle, CUSPARSE_OPERATION_NON_TRANSPOSE, &alpha, matrix_, vecX,
       &beta, vecY, CUDA_R_32F, CUSPARSE_SPMV_CSR_ALG1, &bufferSize);
+  printf("spmv bufferSize: %zu\n", bufferSize);
 
   void *dBuffer = NULL;
   CUDAContext::get_instance().make_current();
   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,
       &beta, vecY, CUDA_R_32F, CUSPARSE_SPMV_CSR_ALG1, dBuffer);