Skip to content

Commit 4e8eb18

Browse files
authored
fix return type of cuBLAS (#749)
1 parent a0e99a3 commit 4e8eb18

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

csrc/activation.cu

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
*/
1616
#include <flashinfer/activation.cuh>
1717

18-
#include "aot_extension_utils.h"
18+
#include "pytorch_extension_utils.h"
1919

2020
using namespace flashinfer;
2121

include/flashinfer/gemm/bmm_fp8.cuh

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
#else
4646
#define FLASHINFER_CUBLAS_CALL(EXPR) \
4747
{ \
48-
cudaError_t e = (EXPR); \
48+
cublasStatus_t e = (EXPR); \
4949
if (e != CUBLAS_STATUS_SUCCESS) { \
5050
return e; \
5151
} \

0 commit comments

Comments
 (0)