CI: 03/19/25 upstream sync #299
rocm-repo-management-api-2
merged
23 commits into
rocm-main
from
ci-upstream-sync-152_1
Added exit 1 if git patch is failed + other checks
9145d617
jax.lax: improve docs for several APIs
8b46e53a
Make blocked_fold_in consistent when the block sizes induce padding
13541e9f
Merge pull request #27198 from jakevdp:lax-docs
30941480
[Pallas TPU] Hoist prologue and epilogue outside of pipeline loop
7c5871f4
`set_mesh` should return the prev_mesh instead of nothing. Users can …
a5c0f200
Merge pull request #27094 from vfdev-5:fix-tsan-numpy-install-patch
ee0073e6
Remove //jaxlib:cpu_kernels and //jaxlib:gpu_kernels forwarding Bazel…
547d6027
[Mosaic GPU] Enable the new transform inference pass in the warpgroup…
875099b2
Adds option to initialize buffers to NaNs or zeros in TPU interpret m…
47e8effd
fix to ragged_all_to_all transpose
942ff38e
Run the stream annotation tests on 2 devices so that it can be tested…
76d9890b
[Mosaic GPU] Support reads/writes from SMEM to WGMMARowFragLayout arr…
54691b12
Fix logging_test fails on Linux with NVIDIA Driver only.
080804c7
Support tuples in custom_partitioning.
0fb59747
Better mosaic lowering for dynamic shapes, extend an interpreter into…
01a110c4
Move jaxlib/{cuda,rocm}_plugin_extension into jaxlib/{cuda/rocm}/
3f91b4b4
Check the type of mesh in `use_abstract_mesh` and `use_concrete_mesh`
663ef7ae
Update XLA dependency to use revision
8c7a55ea
Make sure to DCE read effects
4d715753
[Pallas/Fuser] DCE fusion jaxprs before pulling (to avoid unnecessary…
e949effc
Integrate LLVM at llvm/llvm-project@0230d63b4a8b
f3b7c5cb
Merge pull request #27227 from jburnim:jburnim_pallas_interpret_mode4
e9ce8fb9
JehandadKhan
approved these changes
on 2025-03-19
Login to write a write a comment.
Login via GitHub