CI: 02/11/25 upstream sync #229
github-actions
merged
41 commits into
rocm-main
from
ci-upstream-sync-112_1
[better_errors] Continue adding debug info to Jaxprs (step 5)
000b92f5
refactor: move lax_numpy window functions into their own file
17215177
[better_errors] Continue adding debug info to Jaxprs (step 4)
1e813e16
Document behavior of linalg.solve in case the system matrix is singular
af794721
[better_errors] Continue adding debug info to Jaxprs (step 7)
817b3e57
[Pallas] Add pipeline mode to pltpu
6740165e
#sdy enable `test_mem_kind_donation_pinned_host` for Shardy.
6ed4c29c
Update XLA dependency to use revision
1402220e
Bump the minimum Mac OS X version for x86 builds to 11.0.
f6ca6866
Merge pull request #26436 from hawkinsp:minver2
91c9bbfa
Add a missing jaxlib version check in ragged_collective_test
26d8e112
Bump oldest supported libtpu to match the compatibility window (12 we…
c24a3a77
Merge pull request #26411 from jakevdp:jnp-window-functions
260a879b
Merge pull request #26384 from gnecula:debug_info_jaxpr_4
1a8d5377
Some lax.linalg housekeeping.
154e4506
Merge pull request #26386 from gnecula:debug_info_jaxpr_5
4f979b44
[pallas] fix typo
e6f38b7f
[TPU][Pallas][XLA] Add BUILD time codegen tool that turns a pallas ke…
6bedabd3
[pallas:triton] Updated :export_pallas_test following the changes in …
700298fd
Merge pull request #26426 from NKlug:nklug-linalg-solve-docu
6a638ac8
[StableHLO] Allow composites to use dtype values in attributes
319f7f5a
Fix missing batch partitioning hook in GPU plugin.
ba8c3a92
Merge pull request #26423 from gnecula:debug_info_jaxpr_7
b7d01228
Add `--config=cuda_libraries_from_stubs` in the end of all additional…
e64650e2
[TPU[Mosaic] Fix missing sfences in smem DMAs
ffd3faad
Re-enable oss paged attn kernel
d3ed6ca0
Fix Python version test against Python 3.12.
1e447c8a
#jax Optimize `jax.numpy.take_along_axis` along the dimension satisfies
4b1400db
Fix debug_info_test.py regexp for Windows paths.
872b8c3a
Merge pull request #26450 from gnecula:debug_info_win2
e7acb20e
[better_errors] Continue adding debug info to Jaxprs (step 6)
550d1aa1
Disable XLA autotuning to speed up tests
c171fc60
Merge pull request #26399 from gnecula:debug_info_jaxpr_6
49eccd6c
Update XLA dependency to use revision
5a223516
[Mosaic GPU] Handle TMEM allocation in the compiler
0209eee1
[Mosaic GPU] Add support for 2-CTA MMA on Blackwell
849ea268
Add "sequential_unrolled" vmap method for callbacks.
c502332e
[pallas:triton] Lift dot_general restriction on minimal tile size for a.
2c165bff
[Mosaic GPU] Add support for non-multicast `.cta_group::2` async_copies
21598d02
[Mosaic GPU] Add support for collective MMA in the Blackwell matmul e…
74e86bab
Merge branch 'rocm-main' into ci-upstream-sync-112_1
3745591d
charleshofer
approved these changes
on 2025-02-11
Login to write a write a comment.
Login via GitHub