File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments