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 <typename FPTYPE>
 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 <typename FPTYPE>
 void memset_device_memory(
   FPTYPE * device,
-  const FPTYPE var,
+  const int var,
   const int size)
   {
     DPErrcheck(hipMemset(device,var,sizeof(FPTYPE)*size));