Add <cstring> header file when using memset() function to avoid compi…

…lation errors. (#4380) Co-authored-by: liuxiaohui <[email protected]>
deepmodeling · Jun 14, 2024 · 46627fc · 46627fc
1 parent 6bf333d
commit 46627fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/module_hamilt_lcao/module_gint/kernels/cuda/cuda_tools.cu b/source/module_hamilt_lcao/module_gint/kernels/cuda/cuda_tools.cu
@@ -1,4 +1,5 @@
 #include <iostream>
+#include <cstring>
 
 #include "cuda_tools.cuh"
 
@@ -266,4 +267,4 @@ inline T* Cuda_Mem_Wrapper<T>::get_host_pointer(int stream_id)
 }
 template class Cuda_Mem_Wrapper<double>;
 template class Cuda_Mem_Wrapper<double*>;
-template class Cuda_Mem_Wrapper<int>;
+template class Cuda_Mem_Wrapper<int>;