From db8f7de4ffea069a5dea5023a1c662969e2dda9f Mon Sep 17 00:00:00 2001
From: Helen Schottenhamml <helen.schottenhamml@fau.de>
Date: Mon, 23 Sep 2024 09:52:58 +0200
Subject: [PATCH] Extend GPU commands

---
 src/gpu/DeviceWrapper.h |  7 +++++++
 src/gpu/GPUWrapper.h    | 14 ++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/gpu/DeviceWrapper.h b/src/gpu/DeviceWrapper.h
index 64590bd4..3bdf1be0 100644
--- a/src/gpu/DeviceWrapper.h
+++ b/src/gpu/DeviceWrapper.h
@@ -146,6 +146,13 @@ inline gpuError_t gpuMemcpyAsync(void* /*dst*/, const void* /*src*/, size_t /*co
 inline gpuError_t gpuMemcpy3D(const gpuMemcpy3DParms* /*p*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
 inline gpuError_t gpuMemcpy3DAsync(const gpuMemcpy3DParms* /*p*/, gpuStream_t /*stream*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
 
+inline gpuError_t gpuMemset(void* /*devPtr*/, int /*value*/, size_t /*count*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+inline gpuError_t gpuMemsetAsync(void* /*devPtr*/, int /*value*/, size_t /*count*/, gpuStream_t /*stream*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+inline gpuError_t gpuMemset2D(void* /*devPtr*/, size_t /*pitch*/, int /*value*/, size_t /*width*/, size_t /*height*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+inline gpuError_t gpuMemset2DAsync(void* /*devPtr*/, size_t /*pitch*/, int /*value*/, size_t /*width*/, size_t /*height*/, gpuStream_t /*stream*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+inline gpuError_t gpuMemset3D(gpuPitchedPtr /*pitchedDevPtr*/, int /*value*/, gpuExtent /*extent*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+inline gpuError_t gpuMemset3DAsync( gpuPitchedPtr /*pitchedDevPtr*/, int  /*value*/, gpuExtent /*extent*/, gpuStream_t /*stream*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
+
 inline gpuPos make_gpuPos(size_t /*x*/, size_t /*y*/, size_t /*z*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
 inline gpuPitchedPtr make_gpuPitchedPtr (void* /*d*/, size_t /*p*/, size_t /*xsz*/, size_t /*ysz*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
 inline gpuExtent make_gpuExtent(size_t /*w*/, size_t /*h*/, size_t /*d*/) { WALBERLA_DEVICE_FUNCTION_ERROR }
diff --git a/src/gpu/GPUWrapper.h b/src/gpu/GPUWrapper.h
index 48fcc2e1..d4893da8 100644
--- a/src/gpu/GPUWrapper.h
+++ b/src/gpu/GPUWrapper.h
@@ -46,6 +46,13 @@
     #define gpuMemcpy3DParms cudaMemcpy3DParms
     #define gpuMemcpy3DAsync cudaMemcpy3DAsync
 
+    #define gpuMemset cudaMemset
+    #define gpuMemsetAsync cudaMemsetAsync
+    #define gpuMemset2D cudaMemset2D
+    #define gpuMemset2DAsync cudaMemset2DAsync
+    #define gpuMemset3D cudaMemset3D
+    #define gpuMemset3DAsync cudaMemset3DAsync
+
     #define make_gpuPos make_cudaPos
     #define make_gpuPitchedPtr make_cudaPitchedPtr
     #define gpuPitchedPtr cudaPitchedPtr
@@ -104,6 +111,13 @@
     #define gpuMemcpy3DParms hipMemcpy3DParms
     #define gpuMemcpy3DAsync hipMemcpy3DAsync
 
+    #define gpuMemset hipMemset
+    #define gpuMemsetAsync hipMemsetAsync
+    #define gpuMemset2D hipMemset2D
+    #define gpuMemset2DAsync hipMemset2DAsync
+    #define gpuMemset3D hipMemset3D
+    #define gpuMemset3DAsync hipMemset3DAsync
+
     #define make_gpuPitchedPtr make_hipPitchedPtr
     #define make_gpuPos make_hipPos
     using gpuPitchedPtr = hipPitchedPtr;