CI: 11/14/24 upstream sync #142
charleshofer
merged
63 commits into
rocm-main
from
ci-upstream-sync-25_1
Re-enable cudnn_fusion_test on A100.
e083c080
Internal: create decorators for defining ufuncs
9b562158
Allow setting default_device with platform names.
afa518aa
jax.scipy.linalg.toeplitz: support implicit batching
3f98c57f
Update some outdated syntax in FFI tutorial.
f7570542
Add an example FFI call to demonstrate the use of global state.
f0864836
Add an option to specify mock GPU topology
eedd0111
Add GPU overflow bugfix (#24846) to changelog.
5808170a
Merge pull request #24860 from dfm:overflow-changelog
b185e64a
Fix pgle profiling, broken in previous change.
64fcb9d3
Merge pull request #24806 from jakevdp:ufunc-decorator
3a5ac487
Specialize ufunc.reduce for monoidal binary ufuncs.
6e1aa3c1
Change to internal dead code elimination. Now the functions in `dce_r…
310ff734
[Mosaic] Fix mask creation for packed sublanes
1221da84
[mosaic_gpu] The profiler now uses FFI calls for creating events and …
d304025a
Merge pull request #24792 from dfm:ffi-tutorial-outdated
8e224dbc
Update XLA dependency to use revision
c4a0369f
Change the assumed width of the bool packing in the early-lowering ch…
370c4a70
[Mosaic] Add parameter names to tpu.sem_signal and add tests
c32db46e
Dedent your yields!
d47e2541
Merge pull request #24842 from jakevdp:batched-toeplitz
ed9fdbbf
Add new CI scripts for running Bazel CPU presubmits
195d4070
[XLA] Make our LLVM usage more googley
f2a25cc2
Fix parallel pgle-tests execution.
9a28b561
Add a shard map replication rule for cond_p.
dfabcb02
Avoid using a contextmanager in Primitive.bind.
bc82203a
Fix bug where the Python wrapper to ParseArguments() didn't intern th…
be3c8be1
[AutoPGLE] Fix test after pjrt cache refactoring
93a1f9d3
Merge pull request #24877 from dfm:cond-rep-rule
d83517ba
Add square_p
4d0a007d
Add Pallas Triton lowering for jax.lax.bitcast_convert_type.
558ebb9f
When caching is enabled, also enable XLA caching features as well
a79d307a
Merge pull request #24069 from sergachev:cudnn_fusion_test_a100
12c8c68c
Merge pull request #22899 from trevor-m:cache
0a755aec
doc: link directly to installation on the main page
72a4692b
Fix typos: Change 'arugments' to 'arguments'.
307e88f2
Merge pull request #24874 from pearu:pearu/square_p
14e08aa2
Update XLA dependency to use revision
4a884d41
Stop using generators for linear_util transformations.
1c9b23c5
Don't override --xla_tpu_use_enhanced_launch_barrier if explicitly set
94f9a488
Merge pull request #24883 from jakevdp:doc-installation
a792fb06
Merge pull request #24886 from carlosgmartin:fix_typos
426e13a5
Avoid repeatedly rebuilding a tuple in issubdtype.
2463bf1f
Merge pull request #24810 from jakevdp:bitwise-reduce
842d93e1
Merge pull request #24873 from jax-ml:no-gen-linear-util
d69fe1f8
Merge pull request #24889 from skye:barrier
268c86f8
Merge pull request #23805 from dfm:ffi-examples-state
83700828
[pallas:mosaic_gpu] Ported two pipelining optimizations to `emit_pipe…
aefe6215
Make the jaxpr for jnp.pad in "constant" mode more succinct.
ad5a0621
Don't use an out-of-line lowering for integer_pow for small powers.
081eaeaa
Merge pull request #23374 from jaro-sevcik:mock-topology-config
a1eb5cea
Fixed typo in asan.yaml
6644bbc9
finalize deprecation of jax.clear_backends
f401c979
Merge pull request #24899 from vfdev-5:patch-2
3380a211
Fix pgle test handling of directory removal.
de07be5c
Merge pull request #24865 from jakevdp:dep-clear-backends
89f411ab
[Mosaic][TPU] Omit short circuiting of relayout (we should always rel…
a8464ce7
[sharding_in_types] Support shard_map with sharding in types. Right n…
05716b58
Merge pull request #24751 from Stella-S-Yan:feature/default_device_str
cea81767
Merge pull request #24896 from hawkinsp:pad
fcde8aa6
Merge pull request #24897 from hawkinsp:ipow
19a51de2
Merge remote-tracking branch 'origin/rocm-main' into ci-upstream-sync…
9426b09c
charleshofer
approved these changes
on 2024-11-14
Add commit to see if it triggers CI
34d9633b
Login to write a write a comment.
Login via GitHub