fix memcpy_device_to_host (rocm)

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Sep 17, 2023 · 664f9ea · 664f9ea
1 parent b6825bb
commit 664f9ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/lib/include/gpu_rocm.h b/source/lib/include/gpu_rocm.h
@@ -59,7 +59,7 @@ void memcpy_host_to_device(FPTYPE *device, const FPTYPE *host, const int size) {
 }
 
 template <typename FPTYPE>
-void memcpy_device_to_host(FPTYPE *device, std::vector<FPTYPE> &host) {
+void memcpy_device_to_host(const FPTYPE *device, std::vector<FPTYPE> &host) {
   DPErrcheck(hipMemcpy(&host[0], device, sizeof(FPTYPE) * host.size(),
                        hipMemcpyDeviceToHost));
 }