diff --git a/src/USER-CUDA/atom_vec_angle_cuda.cpp b/src/USER-CUDA/atom_vec_angle_cuda.cpp index b816b690b1..08ba9761ed 100644 --- a/src/USER-CUDA/atom_vec_angle_cuda.cpp +++ b/src/USER-CUDA/atom_vec_angle_cuda.cpp @@ -127,7 +127,11 @@ void AtomVecAngleCuda::grow_send(int n,double** buf_send,int flag) //need to be void AtomVecAngleCuda::grow_both(int n) { if(cuda->finished_setup) - cuda->downloadAll(); + { + cuda->cu_special->upload(); + cuda->cu_nspecial->upload(); + cuda->downloadAll(); + } AtomVecAngle::grow(n); if(cuda->finished_setup) { diff --git a/src/USER-CUDA/atom_vec_full_cuda.cpp b/src/USER-CUDA/atom_vec_full_cuda.cpp index 9e656ed58d..9816a7c5c9 100644 --- a/src/USER-CUDA/atom_vec_full_cuda.cpp +++ b/src/USER-CUDA/atom_vec_full_cuda.cpp @@ -127,7 +127,11 @@ void AtomVecFullCuda::grow_send(int n,double** buf_send,int flag) //need to be void AtomVecFullCuda::grow_both(int n) { if(cuda->finished_setup) - cuda->downloadAll(); + { + cuda->cu_special->upload(); + cuda->cu_nspecial->upload(); + cuda->downloadAll(); + } AtomVecFull::grow(n); if(cuda->finished_setup) {