diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0b349eb394..a1c977cd4c 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -499,7 +499,7 @@ function(MFC_SETUP_TARGET)
                 endforeach()
 
                 target_compile_options(${a_target}
-                    PRIVATE -gpu=keep,ptxinfo,lineinfo
+                    PRIVATE -gpu=keep,ptxinfo,lineinfo -gpu=maxregcount:128
                 )
 
                 if (MFC_Fastmath)
diff --git a/src/simulation/m_igr.fpp b/src/simulation/m_igr.fpp
index 0fbc76346f..c7b35d5808 100644
--- a/src/simulation/m_igr.fpp
+++ b/src/simulation/m_igr.fpp
@@ -27,6 +27,7 @@ module m_igr
 #ifdef __NVCOMPILER_GPU_UNIFIED_MEM
     integer, dimension(3) :: nv_uvm_temp_on_gpu
     real(wp), pointer, contiguous, dimension(:, :, :) :: jac, jac_rhs, jac_old
+    $:GPU_DECLARE(create='[jac, jac_rhs, jac_old]')
     real(wp), allocatable, dimension(:, :, :), pinned, target :: jac_host
     real(wp), allocatable, dimension(:, :, :), pinned, target :: jac_rhs_host
     real(wp), allocatable, dimension(:, :, :), pinned, target :: jac_old_host