Skip to content

Commit a7aee47

Browse files
authored
ggml-cuda: Fix HIP build (#4528)
regression of #4490 Adds defines for two new datatypes cublasComputeType_t, cudaDataType_t. Currently using deprecated hipblasDatatype_t since newer ones very recent.
1 parent 0e18b2e commit a7aee47

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml-cuda.cu

+2
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
#define CUDA_R_16F HIPBLAS_R_16F
3232
#define CUDA_R_32F HIPBLAS_R_32F
3333
#define __shfl_xor_sync(mask, var, laneMask, width) __shfl_xor(var, laneMask, width)
34+
#define cublasComputeType_t hipblasDatatype_t //deprecated, new hipblasComputeType_t not in 5.6
3435
#define cublasCreate hipblasCreate
3536
#define cublasGemmEx hipblasGemmEx
3637
#define cublasGemmBatchedEx hipblasGemmBatchedEx
@@ -40,6 +41,7 @@
4041
#define cublasSetStream hipblasSetStream
4142
#define cublasSgemm hipblasSgemm
4243
#define cublasStatus_t hipblasStatus_t
44+
#define cudaDataType_t hipblasDatatype_t //deprecated, new hipblasDatatype not in 5.6
4345
#define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer
4446
#define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess
4547
#define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess

0 commit comments

Comments
 (0)