Update JAX to commit 45065d569064392cee45a066f9f788fe29ee2cd8

[giordano]

Diff: jax-ml/jax@4455434...45065d5

[wsmoses]

@giordano this should get force pushed with the new commit right? [ironically the error here was a bug in jax proper, where we dont need to adapt, but presumably fixed on their end]

[giordano]

Yes, but apparently this failed yesterday: https://github.com/EnzymeAD/Enzyme-JAX/actions/runs/17991295111/job/51181693038. I don't quite understand the error, my hunch is that this was a transient failure in querying github API

[giordano]

Uhm, it failed again. I'll have a look.

[giordano]

The error is

Error: Unable to process file command 'env' successfully.
Error: Invalid format 'Diff: https://github.com/jax-ml/jax/compare/4455434869812dfb7336b18f2e59a8690ba38e79...6c1760b695ee770c18f9a8b2f767398294a2c983'

Getting the commit message is actually working right, the text Diff: https://github.com/jax-ml/jax/compare/4455434869812dfb7336b18f2e59a8690ba38e79...6c1760b695ee770c18f9a8b2f767398294a2c983 comes from 1eb8516. I believe there's something going wrong when writing the multiline commit message into ${GITHUB_ENV}, the line Diff: https://github.com/jax-ml/jax/compare/4455434869812dfb7336b18f2e59a8690ba38e79...6c1760b695ee770c18f9a8b2f767398294a2c983 is somehow "interpreted"?

Edit: resolved by #1424 by avoiding going through ${GITHUB_ENV}.

[giordano]

Error in patch: Error applying patch /__w/Enzyme-JAX/Enzyme-JAX/patches/xla.patch: in patch applied to /root/.bazel/external/xla/xla/backends/gpu/codegen/triton/BUILD: could not apply patch due to CONTENT_DOES_NOT_MATCH_TARGET, error applying change near line 262

[wsmoses]

ah yeah thats [a slightly different] upstream fix to the one we vendored earlier, we can remove our patch

[giordano]

ERROR: /root/.bazel/external/triton/third_party/proton/BUILD: no such target '@@triton//third_party/proton:ProtonIRDialect': target 'ProtonIRDialect' not declared in package 'third_party/proton' defined by /root/.bazel/external/triton/third_party/proton/BUILD
ERROR: /__w/Enzyme-JAX/Enzyme-JAX/src/enzyme_ad/jax/BUILD:784:11: no such target '@@triton//third_party/proton:ProtonIRDialect': target 'ProtonIRDialect' not declared in package 'third_party/proton' defined by /root/.bazel/external/triton/third_party/proton/BUILD and referenced by '//src/enzyme_ad/jax:RegistryUtils'

[wsmoses]

wat: https://github.com/openxla/triton/blob/4038c4a545d65b35b43b3866945ab515ce45ab1d/third_party/proton/BUILD#L84