Cleanup JAX/XLA directories before checking out new commits

NVIDIA · Oct 10, 2024 · 363f41c · 363f41c
1 parent e5c0587
commit 363f41c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/triage/jax_toolbox_triage/main.py b/.github/triage/jax_toolbox_triage/main.py
@@ -175,6 +175,8 @@ def build_and_test(
             jaxlib, and run the test command. Throws on error when checking out or
             building, and returns the status of the test command.
             """
+            worker.check_exec(["git", "stash"], workdir=xla_dir)
+            worker.check_exec(["git", "stash"], workdir=jax_dir)
             worker.check_exec(["git", "checkout", xla_commit], workdir=xla_dir)
             worker.check_exec(["git", "checkout", jax_commit], workdir=jax_dir)
             logger.info(f"Checking out XLA {xla_commit} JAX {jax_commit}")