From 5987b7dcb8b37b9339fc64d8af4d48cd122d9f47 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=B3=BD=E5=AE=87?= Date: Mon, 22 Nov 2021 16:19:08 +0800 Subject: [PATCH] fix bugs about parameters of memset --- source/lib/include/gpu_cuda.h | 2 +- source/lib/include/gpu_rocm.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/source/lib/include/gpu_cuda.h b/source/lib/include/gpu_cuda.h index 8a2b617c95..cc23969857 100644 --- a/source/lib/include/gpu_cuda.h +++ b/source/lib/include/gpu_cuda.h @@ -154,7 +154,7 @@ void delete_device_memory( template void memset_device_memory( FPTYPE * device, - const FPTYPE var, + const int var, const int size) { DPErrcheck(cudaMemset(device, var, sizeof(FPTYPE) * size)); diff --git a/source/lib/include/gpu_rocm.h b/source/lib/include/gpu_rocm.h index b6439c3bb8..84e05efbd2 100644 --- a/source/lib/include/gpu_rocm.h +++ b/source/lib/include/gpu_rocm.h @@ -110,7 +110,7 @@ void delete_device_memory( template void memset_device_memory( FPTYPE * device, - const FPTYPE var, + const int var, const int size) { DPErrcheck(hipMemset(device,var,sizeof(FPTYPE)*size));