summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorarlo-phoenix <140345165+arlo-phoenix@users.noreply.github.com>2023-12-18 22:33:45 +0100
committerGitHub <noreply@github.com>2023-12-18 22:33:45 +0100
commita7aee47b98e45539d491071b25778b833b77e387 (patch)
treebdc6030e44c3172e367e0a5a45c9b019a8b97952
parent0e18b2e7d0b5c0a509ea40098def234b8d4a938a (diff)
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.
-rw-r--r--ggml-cuda.cu2
1 files changed, 2 insertions, 0 deletions
diff --git a/ggml-cuda.cu b/ggml-cuda.cu
index d0f3d803..f20846fe 100644
--- a/ggml-cuda.cu
+++ b/ggml-cuda.cu
@@ -31,6 +31,7 @@
#define CUDA_R_16F HIPBLAS_R_16F
#define CUDA_R_32F HIPBLAS_R_32F
#define __shfl_xor_sync(mask, var, laneMask, width) __shfl_xor(var, laneMask, width)
+#define cublasComputeType_t hipblasDatatype_t //deprecated, new hipblasComputeType_t not in 5.6
#define cublasCreate hipblasCreate
#define cublasGemmEx hipblasGemmEx
#define cublasGemmBatchedEx hipblasGemmBatchedEx
@@ -40,6 +41,7 @@
#define cublasSetStream hipblasSetStream
#define cublasSgemm hipblasSgemm
#define cublasStatus_t hipblasStatus_t
+#define cudaDataType_t hipblasDatatype_t //deprecated, new hipblasDatatype not in 5.6
#define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer
#define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess
#define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess