Optimize BSSN-EM 8th-order AMR transfers

2026-05-07 21:38:16 +08:00
parent dcc83bafcb
commit 1064a68d16
2 changed files with 241 additions and 0 deletions
--- a/makefile_and_run.py
+++ b/makefile_and_run.py
@@ -167,6 +167,12 @@ def _gpu_runtime_env():
            "AMSS_INTERP_GPU": "0",
            "AMSS_CUDA_AWARE_MPI": "0",
        })
+    if finite_difference == "8th-order" and getattr(input_data, "Equation_Class", "") == "BSSN-EM":
+        defaults.update({
+            "AMSS_CUDA_AMR_RESTRICT_DEVICE": "1",
+            "AMSS_CUDA_AMR_RESTRICT_BATCH": "1",
+            "AMSS_CUDA_DEVICE_SEGMENT_BATCH": "1",
+        })
    if getattr(input_data, "Equation_Class", "") in ("BSSN", "BSSN-EScalar", "Z4C"):
        defaults["AMSS_CUDA_AMR_RESTRICT_DEVICE"] = "1"
    if getattr(input_data, "Equation_Class", "") == "Z4C":