add CupyArrayContext support

examples/ablation-workshop.py

@@ -1212,13 +1212,16 @@ def my_post_step(step, t, dt, state):
                         dest="restart_file", nargs="?", action="store",
                         help="simulation restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
     from warnings import warn
     warn("Automatically turning off DV logging. MIRGE-Com Issue(578)")
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/advection_diffusion_reaction.py

@@ -331,13 +331,16 @@ def my_rhs(t, u):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
     lazy = args.lazy
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(lazy=args.lazy, distributed=True,
                                                     profiling=args.profiling,
-                                                    numpy=args.numpy)
+                                                    numpy=args.numpy,
+                                                    cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/autoignition.py

@@ -657,6 +657,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
     from warnings import warn
     warn("Automatically turning off DV logging. MIRGE-Com Issue(578)")
@@ -672,7 +674,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/blasius.py

@@ -523,6 +523,8 @@ def my_post_step(step, t, dt, state):
         help="enable lazy evaluation [OFF]")
     parser.add_argument("--numpy", action="store_true",
         help="use numpy-based eager actx.")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
 
     args = parser.parse_args()
 
@@ -538,7 +540,8 @@ def my_post_step(step, t, dt, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/combozzle.py

@@ -1255,6 +1255,8 @@ def dummy_rhs(t, state):
     parser.add_argument("--casename", help="casename to use for i/o")
     parser.add_argument("--tpe", action="store_true",
                         help="Use tensor product elements (quads/hexes).")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -1271,7 +1273,8 @@ def dummy_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/doublemach.py

@@ -443,6 +443,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -455,7 +457,7 @@ def my_rhs(t, state):
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(lazy=args.lazy, distributed=True,
                                                     profiling=args.profiling,
-                                                    numpy=args.numpy)
+                                                    numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/doublemach_physical_av.py

@@ -716,6 +716,8 @@ def _my_rhs_phys_visc_div_av(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -730,7 +732,7 @@ def _my_rhs_phys_visc_div_av(t, state):
     actx_class = get_reasonable_array_context_class(lazy=args.lazy,
                                                     distributed=True,
                                                     profiling=args.profiling,
-                                                    numpy=args.numpy)
+                                                    numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/heat-source.py

@@ -245,11 +245,14 @@ def my_post_step(step, t, dt, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/hotplate.py

@@ -448,6 +448,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -460,7 +462,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/lump.py

@@ -383,6 +383,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -395,7 +397,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/mixture.py

@@ -447,6 +447,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
     from warnings import warn
     warn("Automatically turning off DV logging. MIRGE-Com Issue(578)")
@@ -461,7 +463,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/multiple-volumes.py

@@ -392,6 +392,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -404,7 +406,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/orthotropic-diffusion.py

@@ -204,11 +204,14 @@ def my_post_step(step, t, dt, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/poiseuille-multispecies.py

@@ -486,6 +486,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -498,7 +500,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/poiseuille.py

@@ -475,6 +475,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -487,7 +489,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/pulse-mixture.py

@@ -421,6 +421,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -433,7 +435,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/pulse-tpe.py

@@ -348,6 +348,9 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
+
     args = parser.parse_args()
 
     from warnings import warn
@@ -362,7 +365,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/pulse.py

@@ -342,6 +342,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -354,7 +356,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/scalar-advdiff.py

@@ -467,6 +467,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -480,7 +482,7 @@ def my_rhs(t, state):
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(lazy=args.lazy, distributed=True,
                                                     profiling=args.profiling,
-                                                    numpy=args.numpy)
+                                                    numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/scalar-lump.py

@@ -393,6 +393,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -405,7 +407,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename:

examples/sod.py

@@ -391,6 +391,8 @@ def my_rhs(t, state):
         help="use numpy-based eager actx.")
     parser.add_argument("--restart_file", help="root name of restart file")
     parser.add_argument("--casename", help="casename to use for i/o")
+    parser.add_argument("--cupy", action="store_true",
+        help="use cupy-based eager actx.")
     args = parser.parse_args()
 
     from warnings import warn
@@ -403,7 +405,8 @@ def my_rhs(t, state):
 
     from mirgecom.array_context import get_reasonable_array_context_class
     actx_class = get_reasonable_array_context_class(
-        lazy=args.lazy, distributed=True, profiling=args.profiling, numpy=args.numpy)
+        lazy=args.lazy, distributed=True, profiling=args.profiling,
+        numpy=args.numpy, cupy=args.cupy)
 
     logging.basicConfig(format="%(message)s", level=logging.INFO)
     if args.casename: