HIP: Enable support for ggml_backend_cuda_register_host_buffer (#15615)

This commit is contained in:
uvos
2025-08-27 13:58:54 +02:00
committed by GitHub
parent 1bded5a3b3
commit 47373271f9

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