CI: 11/08/24 upstream sync #135
charleshofer
merged
26 commits into
rocm-main
from
ci-upstream-sync-13_1
[Mosaic GPU] Improve correctness of benchmarking scripts
ca261ac5
Fix debug_nans false positive in jnp.quantile
44c6883c
Adding start index and kv_seq_len to decode kernel
d62510bf
Merge pull request #24733 from jakevdp:fix-quantile
36017a4a
Remove some obsolete deprecation registrations
4e45a6d9
register_dataclass: allow marking static fields via field(static=True)
1af3b01c
Update XLA dependency to use revision
dc33a280
Merge pull request #24602 from rdyro:rdyro-decode-attention-mask
37af1002
Merge pull request #24645 from andportnoy:mosaic-gpu-improve-benchmar…
4cc80889
[Mosaic GPU] Fix the ordering of transforms in async_copy
50667129
[Pallas:MGPU] Add support for multiple heads in attention
f8dba3c8
[Mosaic GPU] Introduce a more flexible layout system
de06584d
[Pallas] Fix lowering tests for reduction ops
1a544b6f
Update XLA dependency to use revision
37fc8349
Reverts 1a544b6f363fbb03edc40e03d759cd42a6b64733
88a62a45
Merge pull request #24700 from jakevdp:register-dataclass
0b022aa7
[Mosaic] Fix canonicalize_extract op name.
8b7bcade
Remove sharding from custom_root_test.
3b2e4a16
[Mosaic] Fix handling of i1 splat constants
04a66522
Fix copy-paste typo
eb2dd2ab
Plumb dot dimension numbers into TPU matmul op.
8b89adc6
Propagate CopySemantics from python to C++ transfer APIs so that devi…
0bb30f07
[Mosaic GPU] Implement the skeleton of a lowering pass for the Mosaic…
1f1d27de
Adding support for copy_p primitive to jet.
6e8a35f0
Add note on etils requirement for the Jax compilation cache.
60a6cd47
Merge branch 'rocm-main' into ci-upstream-sync-13_1
ced1e2ba
charleshofer
approved these changes
on 2024-11-12
Login to write a write a comment.
Login via GitHub