Skip to content

Commit d415669

Browse files
committed
cuda : add ROCm / hipBLAS cublasGemmBatchedEx define
1 parent 878aa4f commit d415669

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ggml-cuda.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
#define __shfl_xor_sync(mask, var, laneMask, width) __shfl_xor(var, laneMask, width)
3030
#define cublasCreate hipblasCreate
3131
#define cublasGemmEx hipblasGemmEx
32+
#define cublasGemmBatchedEx hipblasGemmBatchedEx
3233
#define cublasHandle_t hipblasHandle_t
3334
#define cublasSetMathMode(handle, mode) CUBLAS_STATUS_SUCCESS
3435
#define cublasSetStream hipblasSetStream

0 commit comments

Comments
 (0)