Skip to content

Comments

Add support for bisecting JAX workflows#134

Open
nitins17 wants to merge 3 commits intofeature/culprit-finderfrom
srnitin/jax-bisection-support
Open

Add support for bisecting JAX workflows#134
nitins17 wants to merge 3 commits intofeature/culprit-finderfrom
srnitin/jax-bisection-support

Conversation

@nitins17
Copy link
Contributor

This PR adds support for bisecting JAX-style workflows. JAX's continuous workflows clone openxla/xla at HEAD. Re-running a failed workflow would require pinning both XLA and JAX commits to the original state.

This is solved by:

  1. Detecting when a bisected project (e.g., jax-ml/jax) requires dependency pinning.
  2. Looking up the commit timestamp of the JAX commit being tested.
  3. Querying the dependency repo (openxla/xla) for the last commit before that timestamp.
  4. Injecting that historical dependency SHA into the workflow inputs (e.g., xla-commit).

Changes
culprit_finder.py

  • Added PROJECT_CONFIG to map repositories to their external dependencies and input variable names.
  • Updated _test_commit to fetch the commit date and resolve the correct dependency SHA before triggering the workflow.

github_client.py

  • Added get_commit(sha) and get_last_commit_before(repo, date)

@nitins17 nitins17 changed the base branch from main to feature/culprit-finder February 11, 2026 19:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant