CI: 12/23/24 upstream sync #190
github-actions
merged
52 commits into
rocm-main
from
ci-upstream-sync-68_1
[ROCm] ci build and dockerfile changes
bfcace49
Remove internal uses of api_util.shaped_abstractify
5dc37d3f
Add support for N-D FFTs with D>3.
c6131ee5
Update documentation and add setup.py pypi bindings
fefebebe
Add experimental support for building JAX CPU and GPU wheels with GCC.
c78ca042
Merge pull request #25148 from ROCm:ci_install_sys_libs-upstream
25713f95
Merge pull request #25440 from ROCm:jax_pypi-upstream
bad6f0f0
Merge pull request #25606 from dfm:fft-nd
dc8b786d
Update XLA dependency to use revision
1b385f38
DOC: clarify API compatibility discussion & mention EffVer
26930366
Merge pull request #25614 from jakevdp:dep-shaped-abstractify
3e2f2aab
Migrate shardy dialect extension to nanobind.
9c3365fb
add test for partial-auto ppermute
9f42b99a
[Mosaic GPU] Prototype of a warp-specialized pipeline emitter for Mos…
d1294385
[Mosaic:TPU] For trunc, expand supported tilings, offsets and bitwidths
307c8d3a
[Mosaic TPU] Add support for sqrt and rsqrt in bf16 on TPUv6
de8fa8fd
Ensure that the two offsets of a dynamic_slice have the same dtype re…
16712b51
Use common set of build options when building jaxlib+plugin artifacts…
6b096b0c
[Mosaic] Remove TODOs that are already addressed or obsolete.
77f3c114
[Mosaic] Extend macros to handle parentheses.
60ebde89
Add more input validation to jax.distributed.initialize.
a24e7032
Delete unused internal symbols
482a6e73
Merge pull request #25489 from emilyfertig:distributed-init-input-val…
64d3871e
add mutable array ref error checks to cond and custom_vjp
b6482f12
[Mosaic:TPU] Fix trunc infer rule after cl/708011538
8b02884c
[Mosaic TPU] Add relayout-insertion pass and support bitwidth change …
2faf5402
[Mosaic:TPU] Roll back cl/708011538 and cl/708112341
05f3a701
Merge pull request #25625 from mattjj:ref-errors-5
5031b6f5
Relax some test tolerances in N-D FFT tests.
0b190bb6
[pallas:mosaic_gpu] Change the fori tests to also take the while_p pa…
20efbd96
Unify abstractify & shaped_abstractify rules
c560f8e0
Make jax.Arrays a necessary part of the cycle in the GC guard test
d2f937e2
Clarify documentation for output_offsets operand of ragged_all_to_all.
db464b3f
Merge pull request #25616 from jakevdp:abstractify
01e8f889
[Mosaic GPU] Add a lowering for simple `async_load` and `async_store`…
dad23fed
Update XLA dependency to use revision
aa386f85
Migrate _mlir Python binding target to nanobind.
0ff3f144
[Mosaic:TPU] Roll forward of cl/708011538 (expanded trunc support), m…
7ecc9471
Avoid calls to warnings.catch_warnings in JAX core code.
59e5ce22
Add int4/uint4 support to bitcast_convert_type
beee98ab
Merge pull request #25646 from jakevdp:int4-casting
043c2603
Merge pull request #25648 from hawkinsp:warnings3
44d67e13
Support int4/uint4 in jnp.ndarray.view
75f36dc3
Merge pull request #25416 from jakevdp:api-compat
3a35155b
Accelerate deprecation of legacy JAX FFI calling convention.
4216f8fa
Change the namespace name to avoid using `export` c++ keyword on name…
e560c6a4
[JAX] Add a test using inputs with different device orders for a sing…
0e0fc0ac
Move output path to be inside the wheel build command execution loop
0159bead
Merge pull request #25650 from jakevdp:view-int4
1c0dee80
Update XLA dependency to use revision
d28b0dac
[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch …
1719986a
Move ragged tests under a new class.
38747a7a
charleshofer
approved these changes
on 2025-01-02
Login to write a write a comment.
Login via GitHub