Skip to content

Commit

Permalink
improve
Browse files Browse the repository at this point in the history
  • Loading branch information
FantasyVR committed Jun 19, 2022
1 parent 5b8a102 commit a1064e8
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
4 changes: 2 additions & 2 deletions misc/sm_gpu_ndarray.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,6 @@
equal = False
break
if equal:
print("Spmv Results is correct!")
print("Spmv result is correct!")
else:
print("Opps! Spmv Results is wrong.")
print("Opps! Spmv result is wrong.")
2 changes: 1 addition & 1 deletion python/taichi/linalg/sparse_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -228,7 +228,7 @@ def spmv(self, x, y):
>>> x = ti.ndarray(shape=4, dtype=val_dt)
>>> y = ti.ndarray(shape=4, dtype=val_dt)
>>> A = ti.linalg.SparseMatrix(n=4, m=4, dtype=ti.f32)
>>> A.build_from_ndarray_cusparse(row_csr, col_csr, value_csr)
>>> A.build_csr_cusparse(row_csr, col_csr, value_csr)
>>> A.spmv(x, y)
"""
if not isinstance(x, Ndarray) or not isinstance(y, Ndarray):
Expand Down
1 change: 1 addition & 0 deletions taichi/backends/cuda/cuda_driver.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -113,6 +113,7 @@ bool CUSPARSEDriver::load_cusparse() {
}

CUSOLVERDriver::CUSOLVERDriver() {
// TODO: move this to a function to avoid zombie object.
load_lib("libcusolver.so", "cusolver.dll");
}

Expand Down

0 comments on commit a1064e8

Please sign in to comment.