Update exx_cal_energy_op.cu to better support GPU

fix the errors related to the "atomicAdd" function. #6547

Author: ZhouXY-PKU

source/source_pw/module_pwdft/kernels/cuda/exx_cal_energy_op.cu

@@ -2,6 +2,7 @@
 #include "source_psi/psi.h"
 
 #include <thrust/complex.h>
+#include "source_base/module_device/device.h"
 
 namespace hamilt
 {
@@ -81,4 +82,4 @@ struct exx_cal_energy_op<std::complex<FPTYPE>, base_device::DEVICE_GPU>
 
 template struct exx_cal_energy_op<std::complex<float>, base_device::DEVICE_GPU>;
 template struct exx_cal_energy_op<std::complex<double>, base_device::DEVICE_GPU>;
-} // namespace hamilt
+} // namespace hamilt