Skip to content

Commit 629f917

Browse files
authored
cuda : add ROCM aliases for CUDA pool stuff (#3918)
1 parent 51b2fc1 commit 629f917

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

ggml-cuda.cu

+6
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,10 @@
3939
#define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer
4040
#define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess
4141
#define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess
42+
#define cudaDeviceGetMemPool hipDeviceGetMemPool
43+
#define cudaMemPoolAttrReleaseThreshold hipMemPoolAttrReleaseThreshold
44+
#define cudaMemPoolSetAttribute hipMemPoolSetAttribute
45+
#define cudaMemPool_t hipMemPool_t
4246
#define cudaDeviceProp hipDeviceProp_t
4347
#define cudaDeviceSynchronize hipDeviceSynchronize
4448
#define cudaError_t hipError_t
@@ -48,13 +52,15 @@
4852
#define cudaEvent_t hipEvent_t
4953
#define cudaEventDestroy hipEventDestroy
5054
#define cudaFree hipFree
55+
#define cudaFreeAsync hipFreeAsync
5156
#define cudaFreeHost hipHostFree
5257
#define cudaGetDevice hipGetDevice
5358
#define cudaGetDeviceCount hipGetDeviceCount
5459
#define cudaGetDeviceProperties hipGetDeviceProperties
5560
#define cudaGetErrorString hipGetErrorString
5661
#define cudaGetLastError hipGetLastError
5762
#define cudaMalloc hipMalloc
63+
#define cudaMallocFromPoolAsync hipMallocFromPoolAsync
5864
#define cudaMallocHost(ptr, size) hipHostMalloc(ptr, size, hipHostMallocDefault)
5965
#define cudaMemcpy hipMemcpy
6066
#define cudaMemcpy2DAsync hipMemcpy2DAsync

0 commit comments

Comments
 (0)