quickfix

Author: henryleberre

src/simulation/m_mpi_proxy.fpp

@@ -380,13 +380,13 @@ contains
         #:for rdma_mpi in [False, True]
             if (rdma_mpi .eqv. ${'.true.' if rdma_mpi else '.false.'}$) then
                 #:if rdma_mpi
-                    !!$acc data attach(p_send, p_recv)
-                    !!$acc host_data use_device(p_send, p_recv)
+                    !$acc data attach(p_send, p_recv)
+                    !$acc host_data use_device(p_send, p_recv)
 
                     call nvtxStartRange("RHS-COMM-SENDRECV-RDMA")
                 #:else
                     call nvtxStartRange("RHS-COMM-DEV2HOST")
-                    !$acc update host(q_cons_buff_send)
+                    !$acc update host(c_divs_buff_send)
                     call nvtxEndRange
 
                     call nvtxStartRange("RHS-COMM-SENDRECV-NO-RMDA")
@@ -400,25 +400,17 @@ contains
                 call nvtxEndRange ! RHS-MPI-SENDRECV-(NO)-RDMA
 
                 #:if rdma_mpi
-                    !!$acc end host_data
-                    !!$acc end data
-                    !!$acc wait
+                    !$acc end host_data
+                    !$acc end data
+                    !$acc wait
                 #:else
                     call nvtxStartRange("RHS-COMM-HOST2DEV")
-                    !$acc update device(q_cons_buff_recv)
+                    !$acc update device(c_divs_buff_recv)
                     call nvtxEndRange
                 #:endif
             end if
         #:endfor
 
-        if (rdma_mpi) then
-            !!$acc end host_data
-            !!$acc end data
-            !!$acc wait
-        else
-            !$acc update device(c_divs_buff_recv)
-        end if
-
         #:block ITERATE_OVER_BUFFER_REGION_SIDED(dir="mpi_dir", loc="pbc_loc", inner_loops=[("i", 1, "nVars")], pack_v_size="nVars")
             if (bc_id_sfs(mpi_dir, pbc_loc)%sf(exlhs, eylhs, ezlhs)%type >= 0) then
                 c_divs_vf(i)%sf(x, y, z) = c_divs_buff_recv(pack_idr)