lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update cusparse trans

This commit is contained in:
lusz 2024-12-29 17:25:18 +08:00
parent 02c3173def
commit ece4e3efaa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/extern/cuda/cusparse/inc/cusparse_wrapper.h vendored
View File

@ -36,7 +36,7 @@ static inline cudaDataType get_dtype(NanoString dtype) {
return CUDA_R_32F;
}
static inline cusparseOperation_t get_trans_type(bool trans) {
if (trans == true) return CUSPARSE_OPERATION_TRANSPOSE;
if (trans) return CUSPARSE_OPERATION_TRANSPOSE;
else return CUSPARSE_OPERATION_NON_TRANSPOSE;
}
} // jittor