CI: 01/06/25 upstream sync #193
charleshofer
merged
61 commits into
rocm-main
from
ci-upstream-sync-78_1
support head size of 256
307ea87a
Fix a typo in documentation for `pinv` function.
75b56548
[pallas:mosaic_gpu] Updated the lowering following the changes in in …
3e7f4811
[pallas:mosaic_gpu] Reduced duplication between `_ensure_fa` and `_en…
a51d6279
[pallas:triton] Add support for lowering `int4` load.
83e60a96
[mosaic_gpu] Include Mosaic GPU dialect fiels into jaxlib
8987867f
Merge pull request #25667 from superbobry:main
76a51f83
Remove casting from jax.nn.one_hot
cb10710c
Use the right include for gmock and gtest
68ec202d
Merge pull request #24607 from kaixih:support_head_size_256
704185ea
Merge pull request #25662 from liblaf:main
6c85e54a
Internal: use a single registry for abstractify APIs
ccc3a295
Merge pull request #25651 from jakevdp:combine-abstractify
51b51026
Update XLA dependency to use revision
23965b74
changelog: link to api compatibility & python version docs
c206ae7f
Merge pull request #25673 from jakevdp:changelog
10876237
![dependabot[bot]](https://avatars.githubusercontent.com/in/29110?v=4&s=60)
Bump actions/upload-artifact from 4.4.3 to 4.5.0
7da753eb
Finalize deprecation of some symbols from jax.lib.xla_client
40fe4b87
Merge pull request #25669 from jakevdp:undep
c57b49c6
[Mosaic:TPU] Vreg-slice-aligned offset changes with scratch retiling
3c79b98c
Switch `mlir` bindings from `pybind11` to `nanobind`
b8091a43
[Mosaic:TPU] In infer ext rule, avoid assigning offsets outside of ds…
44529609
Merge pull request #25674 from jax-ml:dependabot/github_actions/actio…
fa9c7edf
Merge pull request #25672 from jakevdp:finalize-dep
4eff1316
[pallas:mosaic_gpu] Addressed a todo in `broadcasted_iota` lowering
44333e1c
Update XLA dependency to use revision
64511a19
[AutoPGLE] Explicitly disable command buffers when profiler is used.
b6aead6f
Update XLA dependency to use revision
42a0d555
Fix formatting in the docs for transposing pytrees
008c25a3
Update XLA dependency to use revision
6dbda900
Update XLA dependency to use revision
7ab61b7f
[pallas:mosaic_gpu] Added some runtime type checking to `copy_*` and …
76ccb199
Update XLA dependency to use revision
8eeedd18
Update XLA dependency to use revision
879fa12d
Fixes the random key sharding in shard_map.
97b1faac
Update XLA dependency to use revision
25fff524
Merge pull request #25692 from yliu120:rng_key_sharding
494c1573
Update XLA dependency to use revision
e37ea587
Fix log10 and log2 for large inputs.
50670bd9
Update XLA dependency to use revision
4a6cfebc
tbp nightly instructions
213e1782
Reverts 83e60a9697ec20023f4e11169edf64e910b93031
dbe9ccd6
Don't use x32 mode for pallas_test
7c984c60
Merge pull request #25661 from rdyro:tb-nightly-instructions
04a0fbec
Merge pull request #25706 from pearu:pearu/log10-large
82001ed5
[Mosaic:TPU][NFC] Clean up unused variable
ac817b48
Fix OSS build for the Mosaic GPU dialect
64433435
Merge pull request #25710 from apaszke:mgpu_dialect_fix
68483b8e
Update XLA dependency to use revision
726950b8
Merge pull request #25686 from Mikcl:docs/working-with-pytrees-format…
800f903f
Compute cost-analysis on only one HLO module.
df36c298
[Mosaic] NFC: Pull out vreg related functions to util.
57b21541
jax.debug.print: respect local np.printoptions
33060632
Merge pull request #25713 from jakevdp:debug-printoptions
0f4677bc
Update XLA dependency to use revision
e4278f78
Update XLA dependency to use revision
9af29700
Add SMEM as a supported Pallas output memory space.
54fd738e
Update XLA dependency to use revision
d0a92c5c
Change to trigger CI
a1734fd3
Skip failing tests
307f0db7
Skip one more test
708f48da
charleshofer
approved these changes
on 2025-01-06
disabled auto-merge 1 year ago
Manually disabled by user
Manually disabled by user
Login to write a write a comment.
Login via GitHub