CI: 03/12/25 upstream sync #276
github-actions
merged
31 commits into
rocm-main
from
ci-upstream-sync-144_1
Update `:build_jaxlib` flag to control whether we should add `py_impo…
1aca76fc
[Mosaic GPU] Add support for .cta_group::2 MMA with n=512 on Blackwell
30a9e1b3
jax.disable_jit: ensure while_loop behaves similarly to non-disable_j…
4ae3211e
Skip ASAN tests for the big Mosaic GPU tests
6f7ce9d0
Fix syntax error and typos for composite primitive docstring.
d191927b
Merge pull request #27059 from jakevdp:fix-while-loop
c2c68c01
Add TPU test jobs to the new CI continuous and nightly/release test w…
7ac63552
Fix scipy.special.gammainc/gammaincc evaluation at boundary points
82b2591b
Add NVIDIA wheel requirements only for Linux builds.
0db14aa3
Fix the assumption that pages_per_seq is already a multiple of num_kv…
eff612a3
Remove installation of NVIDIA wheels for CPU tests
e0545a71
Increase the number of iterations in a test that compares rolled vers…
67aa997f
[Mosaic GPU] Replace `WGMMAFragLayout` with `TiledLayout` in the mlir…
99c91060
Merge pull request #20699 from pearu:pearu/gammainc
7ac088c1
Remove unsupported mac x86 CI build options
4df691ec
Upgrade `ml-dtypes` version in `py3.10`-`py3.13` hermetic python lock…
13eb8d3a
[Pallas TPU] Fix preferred_element_type propagation in dot_general wi…
29bfd00f
Fix a bug where `full` and `use_mesh` outside jit did not work becaus…
f45cbf33
[Pallas/Fuser] Add custom evaluate to allow/disallow transposes
c6b164dc
Rename `get_ty` to `typeof` which is an alias of `get_aval`
3a26804c
add autodiff rules for jax.lax.ragged_all_to_all collective
66a6eb29
Run single python version for v4-8 and min & max for v5e-8 for TPU te…
ff751ecc
Add support for scratch buffers in `jax_triton`.
74b4d868
Update XLA dependency to use revision
61ba2b26
Ignore Pallas TPU tests when testing with the oldest supported libtpu
a6ab6bbc
Fix matrix exclude syntax in TPU tests block
d89835ac
[pallas:mosaic_gpu] Added support for reductions to the WG lowering
e33f3fc4
Fix integer overflow in workspace size computations for experimental.…
8b7cfcb3
[sharding_in_types] Initial commit to add `varying_manual_axes: froze…
abcc7fdf
Merge remote-tracking branch 'origin/rocm-main' into ci-upstream-sync…
a0edd3fb
Change to run CI
db8ba1b5
charleshofer
approved these changes
on 2025-03-12
Login to write a write a comment.
Login via GitHub