Fix GPU, but so ugly...

Author: Flying-dragon-boxing

source/source_pw/module_pwdft/operator_pw/exx_pw_pot.cpp

@@ -173,7 +173,16 @@ void get_exx_potential(const K_Vectors* kv,
     }
 
     // copy the potential to the device memory
+#ifdef __CUDA
+    cudaError_t err = cudaHostRegister(pot_cpu, sizeof(Real) * npw, cudaHostRegisterPortable);
+    if (err != cudaSuccess) {
+        printf("临时锁定失败: %s\n", cudaGetErrorString(err));
+    }
+#endif
     syncmem_real_c2d_op()(pot, pot_cpu, rhopw_dev->npw);
+#ifdef __CUDA
+    cudaHostUnregister(pot_cpu);
+#endif
 
     delete pot_cpu;
 }
@@ -341,7 +350,16 @@ void get_exx_stress_potential(const K_Vectors* kv,
     }
 
     // copy the potential to the device memory
+#ifdef __CUDA
+    cudaError_t err = cudaHostRegister(pot_cpu, sizeof(Real) * npw, cudaHostRegisterPortable);
+    if (err != cudaSuccess) {
+        printf("临时锁定失败: %s\n", cudaGetErrorString(err));
+    }
+#endif
     syncmem_real_c2d_op()(pot, pot_cpu, rhopw_dev->npw);
+#ifdef __CUDA
+    cudaHostUnregister(pot_cpu);
+#endif
 
     delete pot_cpu;
 }