From 6131035e587c4def63f3872cf686698a7414113d Mon Sep 17 00:00:00 2001 From: shudson Date: Wed, 3 Jan 2024 16:13:20 -0600 Subject: [PATCH] Update Perlmutter platform detection --- libensemble/resources/platforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libensemble/resources/platforms.py b/libensemble/resources/platforms.py index eb521048e..a186d4c09 100644 --- a/libensemble/resources/platforms.py +++ b/libensemble/resources/platforms.py @@ -292,7 +292,7 @@ def known_envs(): """Detect system by environment variables""" platform_info = {} if os.environ.get("NERSC_HOST") == "perlmutter": - if os.environ.get("SLURM_JOB_PARTITION").startswith("gpu_"): + if "gpu_" in os.environ.get("SLURM_JOB_PARTITION"): platform_info = specs_dump(PerlmutterGPU(), by_alias=True) else: platform_info = specs_dump(PerlmutterCPU(), by_alias=True)