|
|
@@ -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
|