@@ -27,9 +27,9 @@ module m_igr
2727#ifdef __NVCOMPILER_GPU_UNIFIED_MEM
2828 integer , dimension (3 ) :: nv_uvm_temp_on_gpu
2929 real (wp), pointer, contiguous, dimension (:, :, :) :: jac, jac_rhs, jac_old
30- real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_host_pool
31- real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_rhs_host_pool
32- real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_old_host_pool
30+ real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_host
31+ real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_rhs_host
32+ real (wp), allocatable, dimension (:, :, :), pinned, target :: jac_old_host
3333#else
3434 real (wp), allocatable, dimension (:, :, :) :: jac, jac_rhs, jac_old
3535 $:GPU_DECLARE(create= ' [jac, jac_rhs, jac_old]' )
@@ -124,21 +124,21 @@ contains
124124 idwbuff(3 )%beg:idwbuff(3 )%end))
125125 @:PREFER_GPU(jac)
126126 else
127- allocate (jac_host_pool (idwbuff(1 )%beg:idwbuff(1 )%end, &
128- idwbuff(2 )%beg:idwbuff(2 )%end, &
129- idwbuff(3 )%beg:idwbuff(3 )%end))
127+ allocate (jac_host (idwbuff(1 )%beg:idwbuff(1 )%end, &
128+ idwbuff(2 )%beg:idwbuff(2 )%end, &
129+ idwbuff(3 )%beg:idwbuff(3 )%end))
130130
131131 jac(idwbuff(1 )%beg:idwbuff(1 )%end, &
132132 idwbuff(2 )%beg:idwbuff(2 )%end, &
133- idwbuff(3 )%beg:idwbuff(3 )%end) = > jac_host_pool (:, :, :)
133+ idwbuff(3 )%beg:idwbuff(3 )%end) = > jac_host (:, :, :)
134134 end if
135135
136136 if (nv_uvm_temp_on_gpu(2 ) == 1 ) then
137137 @:ALLOCATE(jac_rhs(- 1 :m,- 1 :n,- 1 :p))
138138 @:PREFER_GPU(jac_rhs)
139139 else
140- allocate (jac_rhs_host_pool (- 1 :m, - 1 :n, - 1 :p))
141- jac_rhs(- 1 :m, - 1 :n, - 1 :p) = > jac_rhs_host_pool (:, :, :)
140+ allocate (jac_rhs_host (- 1 :m, - 1 :n, - 1 :p))
141+ jac_rhs(- 1 :m, - 1 :n, - 1 :p) = > jac_rhs_host (:, :, :)
142142 end if
143143
144144 if (igr_iter_solver == 1 ) then ! Jacobi iteration
@@ -148,13 +148,13 @@ contains
148148 idwbuff(3 )%beg:idwbuff(3 )%end))
149149 @:PREFER_GPU(jac_old)
150150 else
151- allocate (jac_old_host_pool (idwbuff(1 )%beg:idwbuff(1 )%end, &
152- idwbuff(2 )%beg:idwbuff(2 )%end, &
153- idwbuff(3 )%beg:idwbuff(3 )%end))
151+ allocate (jac_old_host (idwbuff(1 )%beg:idwbuff(1 )%end, &
152+ idwbuff(2 )%beg:idwbuff(2 )%end, &
153+ idwbuff(3 )%beg:idwbuff(3 )%end))
154154
155155 jac_old(idwbuff(1 )%beg:idwbuff(1 )%end, &
156156 idwbuff(2 )%beg:idwbuff(2 )%end, &
157- idwbuff(3 )%beg:idwbuff(3 )%end) = > jac_old_host_pool (:, :, :)
157+ idwbuff(3 )%beg:idwbuff(3 )%end) = > jac_old_host (:, :, :)
158158 end if
159159 end if
160160#endif
@@ -2678,22 +2678,22 @@ contains
26782678 @:DEALLOCATE(jac)
26792679 else
26802680 nullify (jac)
2681- deallocate (jac_host_pool )
2681+ deallocate (jac_host )
26822682 end if
26832683
26842684 if (nv_uvm_temp_on_gpu(2 ) == 1 ) then
26852685 @:DEALLOCATE(jac_rhs)
26862686 else
26872687 nullify (jac_rhs)
2688- deallocate (jac_rhs_host_pool )
2688+ deallocate (jac_rhs_host )
26892689 end if
26902690
26912691 if (igr_iter_solver == 1 ) then ! Jacobi iteration
26922692 if (nv_uvm_temp_on_gpu(3 ) == 1 ) then
26932693 @:DEALLOCATE(jac_old)
26942694 else
26952695 nullify (jac_old)
2696- deallocate (jac_old_host_pool )
2696+ deallocate (jac_old_host )
26972697 end if
26982698 end if
26992699#endif
0 commit comments