From 790ee55c1b2c0ce3f12389abf34ab0091b49aaff Mon Sep 17 00:00:00 2001 From: Patrick Meade Date: Thu, 11 Jan 2024 04:24:29 -0600 Subject: [PATCH] Modify nersc_controller to avoid getting OOM killed --- lta.code-workspace | 7 +++++++ resources/nersc_controller.py | 3 ++- 2 files changed, 9 insertions(+), 1 deletion(-) create mode 100644 lta.code-workspace diff --git a/lta.code-workspace b/lta.code-workspace new file mode 100644 index 0000000..362d7c2 --- /dev/null +++ b/lta.code-workspace @@ -0,0 +1,7 @@ +{ + "folders": [ + { + "path": "." + } + ] +} \ No newline at end of file diff --git a/resources/nersc_controller.py b/resources/nersc_controller.py index 70410db..233853e 100644 --- a/resources/nersc_controller.py +++ b/resources/nersc_controller.py @@ -223,8 +223,9 @@ def get_active_jobs(context: Context) -> JsonObj: # run the squeue command to determine our jobs currently running in the slurm queue # squeue_path The path to the 'squeue' command + # --me Equivalent to --user=. (see only my jobs) # --json Please give me the output in JSON format (easy to parse) - args = [squeue_path, "--json"] + args = [squeue_path, "--me", "--json"] LOG.info(f"Running command: {args}") completed_process = run(args, stdout=PIPE, stderr=PIPE)