From 0e9c74648e1f82d9d524f8cbc4636141710678f3 Mon Sep 17 00:00:00 2001 From: Steboss Date: Tue, 24 Mar 2026 10:36:24 +0000 Subject: [PATCH 1/3] export compilation cache false --- .github/triage/jax_toolbox_triage/pyxis.py | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/triage/jax_toolbox_triage/pyxis.py b/.github/triage/jax_toolbox_triage/pyxis.py index baf6138d9..e9395fcb5 100644 --- a/.github/triage/jax_toolbox_triage/pyxis.py +++ b/.github/triage/jax_toolbox_triage/pyxis.py @@ -73,6 +73,7 @@ def exec( f"--container-name={self._name}", "--container-remap-root", "--no-container-mount-home", + "--export=JAX_ENABLE_COMPILATION_CACHE=false", ] + self._mount_args + policy_args From 7afc4e5282e6d081476353a361113ede1db7aa91 Mon Sep 17 00:00:00 2001 From: Steboss Date: Wed, 25 Mar 2026 14:18:13 +0000 Subject: [PATCH 2/3] prefix jax cache option --- .github/triage/jax_toolbox_triage/pyxis.py | 1 - .github/triage/jax_toolbox_triage/triage_tool.py | 15 +++++++++------ 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/.github/triage/jax_toolbox_triage/pyxis.py b/.github/triage/jax_toolbox_triage/pyxis.py index e9395fcb5..baf6138d9 100644 --- a/.github/triage/jax_toolbox_triage/pyxis.py +++ b/.github/triage/jax_toolbox_triage/pyxis.py @@ -73,7 +73,6 @@ def exec( f"--container-name={self._name}", "--container-remap-root", "--no-container-mount-home", - "--export=JAX_ENABLE_COMPILATION_CACHE=false", ] + self._mount_args + policy_args diff --git a/.github/triage/jax_toolbox_triage/triage_tool.py b/.github/triage/jax_toolbox_triage/triage_tool.py index a75718d2b..8f33a189e 100644 --- a/.github/triage/jax_toolbox_triage/triage_tool.py +++ b/.github/triage/jax_toolbox_triage/triage_tool.py @@ -37,6 +37,11 @@ class TriageTool: def __init__(self, args, logger): self.args = args + # prefix test_command with jax cache option + self.args.test_command = [ + "env", + "JAX_ENABLE_COMPILATION_CACHE=false", + ] + self.args.test_command self.logger = logger self.bisection_url = None self.bisection_versions = {} @@ -73,9 +78,7 @@ def _test_output_directory( ) out_dir = self.args.output_prefix / out_dirname - assert not out_dir.exists(), ( - f"{out_dir} should not already exist, maybe you are re-using {self.args.output_prefix}?" - ) + assert not out_dir.exists(), f"{out_dir} should not already exist, maybe you are re-using {self.args.output_prefix}?" out_dir.mkdir(mode=0o755) return out_dir.resolve() @@ -119,9 +122,9 @@ def _get_versions( """ if explicit_versions is not None and container_url is None: return explicit_versions, None, None, None - assert container_url is not None, ( - "Container URL must be provided if explicit versions are not set." - ) + assert ( + container_url is not None + ), "Container URL must be provided if explicit versions are not set." with self._make_container(container_url) as worker: url_versions, dirs, env = get_versions_dirs_env(worker, versions_from_env) From 4cd3be15ce80b8a8ca9dce7b307bdddae4c421bc Mon Sep 17 00:00:00 2001 From: Steboss Date: Wed, 25 Mar 2026 14:25:44 +0000 Subject: [PATCH 3/3] fix ruff --- .github/triage/jax_toolbox_triage/triage_tool.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/.github/triage/jax_toolbox_triage/triage_tool.py b/.github/triage/jax_toolbox_triage/triage_tool.py index 8f33a189e..ccd13b01a 100644 --- a/.github/triage/jax_toolbox_triage/triage_tool.py +++ b/.github/triage/jax_toolbox_triage/triage_tool.py @@ -78,7 +78,9 @@ def _test_output_directory( ) out_dir = self.args.output_prefix / out_dirname - assert not out_dir.exists(), f"{out_dir} should not already exist, maybe you are re-using {self.args.output_prefix}?" + assert not out_dir.exists(), ( + f"{out_dir} should not already exist, maybe you are re-using {self.args.output_prefix}?" + ) out_dir.mkdir(mode=0o755) return out_dir.resolve() @@ -122,9 +124,9 @@ def _get_versions( """ if explicit_versions is not None and container_url is None: return explicit_versions, None, None, None - assert ( - container_url is not None - ), "Container URL must be provided if explicit versions are not set." + assert container_url is not None, ( + "Container URL must be provided if explicit versions are not set." + ) with self._make_container(container_url) as worker: url_versions, dirs, env = get_versions_dirs_env(worker, versions_from_env)