[pull] master from tensorflow:master#365
Merged
pull[bot] merged 15 commits intoFage2016:masterfrom Apr 25, 2025
Merged
Conversation
PiperOrigin-RevId: 751437289
PiperOrigin-RevId: 751445832
PiperOrigin-RevId: 751453252
…xla/python. Remove uses of the ops module from JAX. JAX no longer uses the XlaBuilder APIs, instead using StableHLO. In passing, move make_tpu_client into xla_bridge so it can more easily import the _profiler module. Expose _jax.approx_top_k_reduction_output_size since we still need that function that we previously obtained via the _ops module. PiperOrigin-RevId: 751470399
PiperOrigin-RevId: 751472968
PiperOrigin-RevId: 751483830
…nd_buffers and recv_buffers as absl::InlinedVector. This makes the lifetime of the container a responsibility of the AllToAll implementation. PiperOrigin-RevId: 751488781
Serialization should not modify the original object, so this CL enforces such a requirement at compile time by marking the serializable as const. PiperOrigin-RevId: 751497675
…ly branches Reducing FFI overheads for jax-ml/jax#26021 ``` name old cpu/op new cpu/op delta BM_AnyBufferArgX1 20.5ns ± 2% 20.4ns ± 1% ~ (p=0.055 n=80+73) BM_AnyBufferArgX4 21.8ns ± 2% 21.7ns ± 1% -0.53% (p=0.006 n=80+69) BM_AsyncAnyBufferArgX1 703ns ± 2% 697ns ± 1% -0.91% (p=0.000 n=77+74) BM_BufferArgX1 20.5ns ± 2% 20.4ns ± 0% -0.68% (p=0.000 n=80+73) BM_BufferArgX4 24.2ns ± 2% 24.9ns ± 0% +3.20% (p=0.000 n=80+70) BM_BufferArgX8 30.8ns ± 2% 30.3ns ± 1% -1.61% (p=0.000 n=79+75) BM_TupleOfI32Attrs 117ns ± 3% 103ns ± 1% -11.89% (p=0.000 n=75+71) BM_EnumAttrs 66.8ns ± 2% 35.5ns ± 1% -46.82% (p=0.000 n=79+73) ``` PiperOrigin-RevId: 751499937
… if next_colocated is not initialized. PiperOrigin-RevId: 751510862
PiperOrigin-RevId: 751528479
… compile time function pointers ``` ---------------------------------------------------------------------- Benchmark Time CPU Iterations ---------------------------------------------------------------------- BM_EnumAttrs 31.1 ns 31.1 ns 22541907 BM_EnumAttrsFunction 37.4 ns 37.4 ns 1874469 BM_EnumAttrsFunctionWrapper 31.3 ns 31.3 ns 22312783 ``` PiperOrigin-RevId: 751532370
PiperOrigin-RevId: 751536187
PiperOrigin-RevId: 751542019
XLA has migrated from `std::string_view` to `absl::string_view`, which supports construction from `nullptr`. The old comments are no longer relevant. PiperOrigin-RevId: 751543245
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
9 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
See Commits and Changes for more details.
Created by
pull[bot] (v2.0.0-alpha.1)
Can you help keep this open source service alive? 💖 Please sponsor : )