Skip to content

Commit 457ae60

Browse files
committed
Replace host_pool with host in m_igr
1 parent a6ff639 commit 457ae60

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

src/simulation/m_igr.fpp

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)