Skip to content

Commit 51d7e90

Browse files
committed
Fix some allocs and deallocs in timesteppers
1 parent b3fdbff commit 51d7e90

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/simulation/m_time_steppers.fpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ contains
115115
end do
116116

117117
#if defined(__NVCOMPILER_GPU_UNIFIED_MEM)
118-
if (nv_uvm_out_of_core) then
118+
if (num_ts == 2 .and. nv_uvm_out_of_core) then
119+
! host allocation for q_cons_ts(2)%vf(j)%sf for all j
119120
allocate(q_cons_ts_pool_host(idwbuff(1)%beg:idwbuff(1)%end, &
120121
idwbuff(2)%beg:idwbuff(2)%end, &
121122
idwbuff(3)%beg:idwbuff(3)%end, &
@@ -1390,7 +1391,9 @@ contains
13901391
end if
13911392
end if
13921393
end do
1393-
deallocate(q_cons_ts_pool_host)
1394+
if (num_ts == 2 .and. nv_uvm_out_of_core) then
1395+
deallocate(q_cons_ts_pool_host)
1396+
end if
13941397
#elif defined(FRONTIER_UNIFIED)
13951398
do i = 1, num_ts
13961399
do j = 1, sys_size
@@ -1407,6 +1410,9 @@ contains
14071410
end do
14081411
end do
14091412
#endif
1413+
do i = 1, num_ts
1414+
@:DEALLOCATE(q_cons_ts(i)%vf)
1415+
end do
14101416

14111417
@:DEALLOCATE(q_cons_ts)
14121418

0 commit comments

Comments
 (0)