Modify nersc_controller to avoid getting OOM killed

WIPACrepo · Jan 11, 2024 · 790ee55 · 790ee55
1 parent 1d39a31
commit 790ee55
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lta.code-workspace b/lta.code-workspace
@@ -0,0 +1,7 @@
+{
+	"folders": [
+		{
+			"path": "."
+		}
+	]
+}
diff --git 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=<my username>. (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)