Add ParmParse parameters to Tutorials/GPU/Launch (AMReX-Codes#1314)

mohuangrui · Aug 26, 2020 · d48f68f · d48f68f
1 parent 655f10a
commit d48f68f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Tutorials/GPU/Launch/main.cpp b/Tutorials/GPU/Launch/main.cpp
@@ -1,6 +1,7 @@
 
 #include <AMReX.H>
 #include <AMReX_MultiFab.H>
+#include <AMReX_ParmParse.H>
 
 #include "MyKernel_F.H"
 
@@ -20,9 +21,14 @@ void main_main ()
 {
     BoxArray ba;
     {
-        Box domain_box(IntVect(0), IntVect(127));
+        int n_cell = 256;
+        int max_grid_size = 64;
+        ParmParse pp;
+        pp.query("n_cell", n_cell);
+        pp.query("max_grid_size", max_grid_size);
+        Box domain_box(IntVect(0), IntVect(n_cell-1));
         ba.define(domain_box);
-        ba.maxSize(64);
+        ba.maxSize(max_grid_size);
     }
 
     MultiFab mf(ba,DistributionMapping{ba},1,0);