diff --git a/src/common/cuda_to_hip.h b/src/common/cuda_to_hip.h
index c12251018..c5c3f90a4 100644
--- a/src/common/cuda_to_hip.h
+++ b/src/common/cuda_to_hip.h
@@ -39,8 +39,8 @@
 
 #define cudaMallocHost                                      hipMallocHost
 #define cudaFreeHost                                        hipFreeHost
-#define cudaMalloc                                          hipMalloc
-#define cudaFree                                            hipFree
+#define cudaMalloc                                          hipMallocHost
+#define cudaFree                                            hipFreeHost
 
 #define cudaMemcpy                                          hipMemcpy
 #define cudaMemcpyAsync                                     hipMemcpyAsync