HIP: Enable support for ggml_backend_cuda_register_host_buffer (llama/15615)

This commit is contained in:
uvos
2025-09-20 13:42:43 +03:00
committed by Georgi Gerganov
parent 02e8b23137
commit 65fa2c0c1a
+1 -1
View File
@@ -3106,7 +3106,7 @@ bool ggml_backend_cuda_register_host_buffer(void * buffer, size_t size) {
return false;
}
#if CUDART_VERSION >= 11010 || defined(GGML_USE_MUSA)
#if CUDART_VERSION >= 11010 || defined(GGML_USE_MUSA) || defined(GGML_USE_HIP)
cudaError_t err = cudaHostRegister(buffer, size, cudaHostRegisterPortable | cudaHostRegisterReadOnly);
if (err != cudaSuccess) {
// clear the error