address the atomicAdd error (#4234)

deepmodeling · May 25, 2024 · 30df1c8 · 30df1c8
1 parent 25584b0
commit 30df1c8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/source/module_hamilt_lcao/module_gint/kernels/cuda/vbatch_matrix_mul.cuh b/source/module_hamilt_lcao/module_gint/kernels/cuda/vbatch_matrix_mul.cuh
@@ -10,6 +10,8 @@
 #include <functional>
 #include "module_cell/unitcell.h"
 #include "module_hamilt_pw/hamilt_pwdft/global.h"
+#include <module_base/module_device/device.h>
+
 
 #define sA(i, j) sA[(j)*slda + (i)]
 #define sB(i, j) sB[(j)*sldb + (i)]

diff --git a/source/module_hamilt_pw/hamilt_pwdft/kernels/cuda/force_op.cu b/source/module_hamilt_pw/hamilt_pwdft/kernels/cuda/force_op.cu
@@ -7,6 +7,7 @@
 #include <thrust/complex.h>
 #include <cuda_runtime.h>
 #include <base/macros/macros.h>
+#include <module_base/module_device/device.h>
 
 #define THREADS_PER_BLOCK 256
 

diff --git a/source/module_hamilt_pw/hamilt_pwdft/kernels/cuda/stress_op.cu b/source/module_hamilt_pw/hamilt_pwdft/kernels/cuda/stress_op.cu
@@ -4,6 +4,7 @@
 #include <complex>
 #include <thrust/complex.h>
 #include <base/macros/macros.h>
+#include <module_base/module_device/device.h>
 
 #include <cuda_runtime.h>