add cudaMemset

LostRuins · Jun 14, 2023 · 61df8e9 · 61df8e9
1 parent a836529
commit 61df8e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ggml-cuda.cu b/ggml-cuda.cu
@@ -50,6 +50,7 @@
 #define cudaMemcpyDeviceToHost hipMemcpyDeviceToHost
 #define cudaMemcpyHostToDevice hipMemcpyHostToDevice
 #define cudaMemcpyKind hipMemcpyKind
+#define cudaMemset hipMemset
 #define cudaOccupancyMaxPotentialBlockSize hipOccupancyMaxPotentialBlockSize
 #define cudaSetDevice hipSetDevice
 #define cudaStreamCreateWithFlags hipStreamCreateWithFlags