@@ -871,22 +871,22 @@ contains
871871 $:END_GPU_PARALLEL_LOOP()
872872
873873 if (qbmm_comm) then
874- #:call GPU_PARALLEL_LOOP_OLD(collapse=5,private=' [r]' )
875- do i = nVar + 1, nVar + 4
876- do l = 0, buff_size - 1
877- do k = -buff_size, n + buff_size
878- do j = -buff_size, m + buff_size
879- do q = 1, nb
880- r = (i - 1) + (q - 1)*4 + v_size* &
881- ((j + buff_size) + (m + 2*buff_size + 1)* &
882- ((k + buff_size) + (n + 2*buff_size + 1)*l))
883- buff_send(r) = real(pb_in(j, k, l + pack_offset, i - nVar, q), kind=wp)
884- end do
874+ $:GPU_PARALLEL_LOOP(collapse=5,private=' [r]' )
875+ do i = nVar + 1, nVar + 4
876+ do l = 0, buff_size - 1
877+ do k = -buff_size, n + buff_size
878+ do j = -buff_size, m + buff_size
879+ do q = 1, nb
880+ r = (i - 1) + (q - 1)*4 + v_size* &
881+ ((j + buff_size) + (m + 2*buff_size + 1)* &
882+ ((k + buff_size) + (n + 2*buff_size + 1)*l))
883+ buff_send(r) = real(pb_in(j, k, l + pack_offset, i - nVar, q), kind=wp)
885884 end do
886885 end do
887886 end do
888887 end do
889- #:endcall GPU_PARALLEL_LOOP_OLD
888+ end do
889+ $:END_GPU_PARALLEL_LOOP()
890890
891891 #:call GPU_PARALLEL_LOOP_OLD(collapse=5,private=' [r]' )
892892 do i = nVar + 1, nVar + 4
0 commit comments