Skip to content

Commit b3fdbff

Browse files
committed
test merge and add nv_uvm_out_of_core back
1 parent 884a4d9 commit b3fdbff

File tree

4 files changed

+10
-4
lines changed

4 files changed

+10
-4
lines changed

src/simulation/m_global_parameters.fpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,7 @@ module m_global_parameters
163163
! 1 => jac on GPU, jac_rhs and jac_old on CPU
164164
! 2 => jac and jac_rhs on GPU, jac_old on CPU
165165
! 3 => jac, jac_rhs, and jac_old on GPU (default)
166+
logical :: nv_uvm_out_of_core ! Enable out-or-core storage of q_cons_ts(2) in timestepping
166167
logical :: nv_uvm_pref_gpu ! Enable explicit gpu memory hints (default TRUE)
167168
!> @}
168169

@@ -584,6 +585,7 @@ contains
584585
585586
! NVIDIA UVM options
586587
nv_uvm_igr_temps_on_gpu = 3 ! => jac, jac_rhs, and jac_old on GPU (default)
588+
nv_uvm_out_of_core = .false.
587589
nv_uvm_pref_gpu = .true.
588590
589591
! Simulation algorithm parameters

src/simulation/m_mpi_proxy.fpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -239,6 +239,7 @@ contains
239239
240240
! NVIDIA UVM variables
241241
call MPI_BCAST(nv_uvm_igr_temps_on_gpu, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
242+
call MPI_BCAST(nv_uvm_out_of_core, 1, MPI_LOGICAL, 0, MPI_COMM_WORLD, ierr)
242243
call MPI_BCAST(nv_uvm_pref_gpu, 1, MPI_LOGICAL, 0, MPI_COMM_WORLD, ierr)
243244
244245
#endif

src/simulation/m_time_steppers.fpp

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -574,7 +574,6 @@ contains
574574
real(wp), intent(inout) :: time_avg
575575

576576
integer :: i, j, k, l, q!< Generic loop iterator
577-
integer :: dest
578577
real(wp) :: start, finish
579578
integer :: dest
580579

@@ -810,9 +809,8 @@ contains
810809
real(wp), intent(INOUT) :: time_avg
811810

812811
integer :: i, j, k, l, q !< Generic loop iterator
813-
integer :: dest
814-
815812
real(wp) :: start, finish
813+
integer :: dest
816814

817815
! Stage 1 of 3
818816

@@ -1385,7 +1383,11 @@ contains
13851383
do j = 1, sys_size
13861384
@:DEALLOCATE(q_cons_ts(1)%vf(j)%sf)
13871385
if (num_ts == 2) then
1388-
nullify(q_cons_ts(2)%vf(j)%sf)
1386+
if (nv_uvm_out_of_core) then
1387+
nullify(q_cons_ts(2)%vf(j)%sf)
1388+
else
1389+
@:DEALLOCATE(q_cons_ts(2)%vf(j)%sf)
1390+
end if
13891391
end if
13901392
end do
13911393
deallocate(q_cons_ts_pool_host)

toolchain/mfc/run/case_dicts.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -314,6 +314,7 @@ def analytic(self):
314314
'ic_eps': ParamType.REAL,
315315
'ic_beta': ParamType.REAL,
316316
'nv_uvm_igr_temps_on_gpu': ParamType.INT,
317+
'nv_uvm_igr_out_of_core': ParamType.LOG,
317318
'nv_uvm_pref_gpu': ParamType.LOG,
318319
})
319320

0 commit comments

Comments
 (0)