Merge branch 'dis' of https://github.com/CaRoLZhangxy/deepmd-kit into…

… dis
deepmodeling · Apr 19, 2024 · 48b9833 · 48b9833
2 parents 06208d2 + a13934b
commit 48b9833
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/op/pt/comm.cc b/source/op/pt/comm.cc
@@ -90,7 +90,7 @@ class Border : public torch::autograd::Function<Border> {
 #endif
 #ifdef GOOGLE_CUDA
         gpuMemcpy(recv_g1, send_g1, nsend * tensor_size * sizeof(FPTYPE),
-                   gpuMemcpyDeviceToDevice);
+                  gpuMemcpyDeviceToDevice);
 #else
       memcpy(recv_g1, send_g1, nsend * tensor_size * sizeof(FPTYPE));
 #endif
@@ -194,7 +194,7 @@ class Border : public torch::autograd::Function<Border> {
         if (nrecv) {
 #ifdef GOOGLE_CUDA
           gpuMemcpy(recv_g1, send_g1, nrecv * tensor_size * sizeof(FPTYPE),
-                     gpuMemcpyDeviceToDevice);
+                    gpuMemcpyDeviceToDevice);
 #else
         memcpy(recv_g1, send_g1, nrecv * tensor_size * sizeof(FPTYPE));
 #endif