HIP: Enable support for ggml_backend_cuda_register_host_buffer (ggml-org#15615)

IMbackK · web-flow · commit 47373271f971 · 2025-08-27T13:58:54.000+02:00
diff --git a/ggml/src/ggml-cuda/ggml-cuda.cu b/ggml/src/ggml-cuda/ggml-cuda.cu
@@ -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

Original file line number	Diff line number	Diff line change
`@@ -3106,7 +3106,7 @@ bool ggml_backend_cuda_register_host_buffer(void * buffer, size_t size) {`
`3106`	`3106`	`return false;`
`3107`	`3107`	`}`
`3108`	`3108`
`3109`		`-#if CUDART_VERSION >= 11010 \|\| defined(GGML_USE_MUSA)`
	`3109`	`+#if CUDART_VERSION >= 11010 \|\| defined(GGML_USE_MUSA) \|\| defined(GGML_USE_HIP)`
`3110`	`3110`	`cudaError_t err = cudaHostRegister(buffer, size, cudaHostRegisterPortable \| cudaHostRegisterReadOnly);`
`3111`	`3111`	`if (err != cudaSuccess) {`
`3112`	`3112`	`// clear the error`