Support 2nd and 8th order CUDA AMR paths
This commit is contained in:
@@ -138,6 +138,7 @@ def _stop_cuda_mps(runtime_env):
|
||||
|
||||
def _gpu_runtime_env():
|
||||
runtime_env = os.environ.copy()
|
||||
finite_difference = str(getattr(input_data, "Finite_Diffenence_Method", "4th-order")).strip()
|
||||
|
||||
defaults = {
|
||||
"AMSS_EVOLVE_TIMING": "1",
|
||||
@@ -160,6 +161,12 @@ def _gpu_runtime_env():
|
||||
"AMSS_CUDA_DEVICE_SEGMENT_BATCH": "0",
|
||||
"AMSS_CUDA_UNCACHED_DEVICE_BUFFERS": "1",
|
||||
}
|
||||
if finite_difference in ("2nd-order", "8th-order"):
|
||||
defaults.update({
|
||||
"AMSS_INTERP_FAST": "0",
|
||||
"AMSS_INTERP_GPU": "0",
|
||||
"AMSS_CUDA_AWARE_MPI": "0",
|
||||
})
|
||||
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":
|
||||
|
||||
Reference in New Issue
Block a user