[pull] master from tensorflow:master - #8790
Merged
Merged
Conversation
The custom layouts used in array_impl/xla_array_impl were not fully canonicalized and was relying on the runtime's implicit layout canonicalization. This change uses a tiled layout for the custom layout tests when running on TPU so that the runtime can apply strong validation for canonicalized layouts and does not have to perform implicit layout canonicalization. PiperOrigin-RevId: 972856960
…le-time constant when compiling under XLA. PiperOrigin-RevId: 972857042
PiperOrigin-RevId: 972867033
PiperOrigin-RevId: 972872123
…tom call legalization - Rename PropagateQsvPass to PropagateQParamsPass across passes definitions, headers, implementations, tests, and converter API registration. - Add backward propagation through TransposeOp and ReshapeOp for per-axis quantization parameters. - Add LegalizeVhloQuantCustomCallsPass to legalize vhlo quant custom calls (quant.dequantize, quant.quantize, quant.fake_quant) to stablehlo.custom_call. PiperOrigin-RevId: 972872299
PiperOrigin-RevId: 972887119
…'s type range to prevent Inf/NaN. During backward constraint propagation, `kConvert` and `kBitcastConvert` operations were passing output constraints through identically to `kCopy` without checking whether the operand has a narrower domain. When downstream 16-bit operations constrained the output to `[-65504, 65504]`, an `f8e4m3fn` operand received this out-of-range interval. Because `f8e4m3fn` has a maximum finite range of `[-448, 448]` and lacks an infinity representation, values outside this range saturated to NaN during random literal generation. This change separates `kConvert` and `kBitcastConvert` in `ConstraintPropagator` to intersect incoming constraints with the finite domain of the operand type, ensuring input parameters are always constrained within their type's representable bounds. PiperOrigin-RevId: 972897290
PiperOrigin-RevId: 972907865
…IRRT IR ops PiperOrigin-RevId: 972917231
PiperOrigin-RevId: 972973724
PiperOrigin-RevId: 972978443
PiperOrigin-RevId: 972982809
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 subscribe to this conversation on GitHub.
Already have an account?
Sign in.
8 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.4)
Can you help keep this open source service alive? 💖 Please sponsor : )