diff options
| author | Kerfuffle <44031344+KerfuffleV2@users.noreply.github.com> | 2023-11-02 13:58:22 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-02 21:58:22 +0200 |
| commit | 629f917cd6b96ba1274c49a8aab163b1b189229d (patch) | |
| tree | 349c686095e65fd6dae3180346b8ee5644f924d0 | |
| parent | 51b2fc11f7f605fff49725a4540e9a6ef7b51b70 (diff) | |
cuda : add ROCM aliases for CUDA pool stuff (#3918)
| -rw-r--r-- | ggml-cuda.cu | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ggml-cuda.cu b/ggml-cuda.cu index 06c28f56..baf02df2 100644 --- a/ggml-cuda.cu +++ b/ggml-cuda.cu @@ -39,6 +39,10 @@ #define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer #define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess #define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess +#define cudaDeviceGetMemPool hipDeviceGetMemPool +#define cudaMemPoolAttrReleaseThreshold hipMemPoolAttrReleaseThreshold +#define cudaMemPoolSetAttribute hipMemPoolSetAttribute +#define cudaMemPool_t hipMemPool_t #define cudaDeviceProp hipDeviceProp_t #define cudaDeviceSynchronize hipDeviceSynchronize #define cudaError_t hipError_t @@ -48,6 +52,7 @@ #define cudaEvent_t hipEvent_t #define cudaEventDestroy hipEventDestroy #define cudaFree hipFree +#define cudaFreeAsync hipFreeAsync #define cudaFreeHost hipHostFree #define cudaGetDevice hipGetDevice #define cudaGetDeviceCount hipGetDeviceCount @@ -55,6 +60,7 @@ #define cudaGetErrorString hipGetErrorString #define cudaGetLastError hipGetLastError #define cudaMalloc hipMalloc +#define cudaMallocFromPoolAsync hipMallocFromPoolAsync #define cudaMallocHost(ptr, size) hipHostMalloc(ptr, size, hipHostMallocDefault) #define cudaMemcpy hipMemcpy #define cudaMemcpy2DAsync hipMemcpy2DAsync |