Skip to content

Update jax-ml/jax to commit 6aaf234b1ee98800fcb80b9014a052a66dadbd8a - #3368

Open
enzymead-bot[bot] wants to merge 1 commit into
mainfrom
update-jax-ml-jax
Open

enzymead-bot[bot] wants to merge 1 commit into
mainfrom
update-jax-ml-jax

Conversation

@enzymead-bot

@enzymead-bot enzymead-bot Bot commented Oct 5, 2026 •

Copy link
Copy Markdown
Contributor

@enzymead-bot enzymead-bot Bot added the dependencies Pull requests that update a dependency file label Oct 5, 2026
@enzymead-bot
enzymead-bot Bot force-pushed the update-jax-ml-jax branch from 8594b4c to dfa35e7 Compare October 7, 2026 01:18
@enzymead-bot enzymead-bot Bot changed the title Update jax-ml/jax to commit 0da15eef9f3a8483d89200772a24b97c73f93b72 Update jax-ml/jax to commit 1e0af75c072ee4a9f3c014b7deec182cc4edec1a Oct 7, 2026
@enzymead-bot
enzymead-bot Bot force-pushed the update-jax-ml-jax branch from dfa35e7 to 5a1d92d Compare October 7, 2026 22:27
@enzymead-bot enzymead-bot Bot changed the title Update jax-ml/jax to commit 1e0af75c072ee4a9f3c014b7deec182cc4edec1a Update jax-ml/jax to commit ad1a9927427f6f305653469f58a7196a5e49c645 Oct 7, 2026
@codecov

codecov Bot commented Oct 7, 2026 •

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 29.92%. Comparing base (0c80202) to head (f4efef0).

Additional details and impacted files
@@           Coverage Diff           @@
##             main    #3368   +/-   ##
=======================================
  Coverage   29.92%   29.92%           
=======================================
  Files         241      241           
  Lines       48628    48628           
=======================================
  Hits        14551    14551           
  Misses      34077    34077           

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@enzymead-bot
enzymead-bot Bot force-pushed the update-jax-ml-jax branch from 5a1d92d to c47a51e Compare October 8, 2026 22:28
@enzymead-bot enzymead-bot Bot changed the title Update jax-ml/jax to commit ad1a9927427f6f305653469f58a7196a5e49c645 Update jax-ml/jax to commit 3a13f7c2788af27aa41fe4f859b6de0aeca6eed3 Oct 8, 2026
@enzymead-bot
enzymead-bot Bot force-pushed the update-jax-ml-jax branch from c47a51e to ad89ef9 Compare October 9, 2026 22:29
@enzymead-bot enzymead-bot Bot changed the title Update jax-ml/jax to commit 3a13f7c2788af27aa41fe4f859b6de0aeca6eed3 Update jax-ml/jax to commit 58846e51f5b2a42a00093f097ca5838855273902 Oct 9, 2026
@wsmoses
wsmoses force-pushed the update-jax-ml-jax branch 3 times, most recently from aa2a215 to 80c50e3 Compare October 10, 2026 17:01
@enzymead-bot
enzymead-bot Bot force-pushed the update-jax-ml-jax branch from 80c50e3 to c6d6f6b Compare October 10, 2026 22:27
@enzymead-bot enzymead-bot Bot changed the title Update jax-ml/jax to commit 58846e51f5b2a42a00093f097ca5838855273902 Update jax-ml/jax to commit 6aaf234b1ee98800fcb80b9014a052a66dadbd8a Oct 10, 2026
Rebase the patches onto the XLA (ed6af492) and LLVM (b08e0059) it pins:
- xla.patch: the rocm BUILD.tpl hunk takes its new context; the
  topology_util.cc hunk (#3376) is upstream and dropped.
- llvm_amdgpu_bb_prolog.patch: new context only; llvm/llvm-project#225433
  is still open.
- llvm_mlir_import_inrange_width.patch (#3367) and
  llvm_orc_unw_revert.patch are upstream and no longer applied.
- The Windows guard appended to stablehlo's temporary.patch is anchored
  where XLA's own InterpreterOps.cpp change now leaves evalRunParallelOp.

And build against that LLVM:
- vendored Polly: IRInserter::InsertHelper no longer overrides, as the
  IRBuilder inserter is no longer virtual (llvm/llvm-project#229700).
- populateGpuToROCDLConversionPatterns takes a ROCDL::TargetInfo in place
  of an amdgpu::Chipset (llvm/llvm-project#223562); the default one is
  the same unknown target.
- raiseimmerse3: the new arith fold of or(x, or(y, x)) changes the
  order affine-cfg reaches the compare in, which is now left as an
  affine.if of true/false rather than turned into a cmpi eq; the same
  happens on main when the input is given that fold already.
- CompareOpCanon: XLA's stablehlo patch gives the reference compareOp
  the op's optional compare type; pass it through.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_016zErYp7upmqr4NHfhod9UD

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

dependencies Pull requests that update a dependency file

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant