[pull] master from tensorflow:master - #8804
Merged
Merged
Conversation
…g_instruction. PiperOrigin-RevId: 975257608
CommonPjRtClientImpl. PiperOrigin-RevId: 975326716
PiperOrigin-RevId: 975336433
Add CuptiTracer::EnableHES() static method to support early initialization of CUPTI activity hardware trace before CUDA context creation. Support enabling HES in GPU tracer via the XPROF_ENABLE_HES environment variable. PiperOrigin-RevId: 975338177
…OpTest - Add execution tests checking component-wise ULP distance with xla::UlpDistance instead of Euclidean relative error. - Enforce <= 2 ULPs for complex64 across all platforms. - Enforce <= 2 ULPs for complex128 on platforms with native float64 units (CPU, GPU), and <= 100 ULPs on TPU due to software-emulated double-double transcendentals. PiperOrigin-RevId: 975342933
Imported from GitHub PR openxla/xla#48140 📝 Summary of Changes Symlink all the runfiles from lit test deps to execution path 🎯 Justification This change is required to support bzlmod builds in lit tests, we now switch rocm ci builds to bzlmod 🚀 Kind of Contribution Please remove what does not apply: ♻️ Cleanup 📊 Benchmark (for Performance Improvements) Not relevant 🧪 Unit Tests: CI 🧪 Execution Tests: CI Copybara import of the project: -- 2ee21fed04c255ec7aeb2ec09215aaa0c2a746e3 by Alexandros Theodoridis <atheodor@amd.com>: Dynamically select rocm runfiles dirs in workspace and bzlmod -- 03e79fa529e898ec2eb2227f88e2958d73eca106 by Alexandros Theodoridis <atheodor@amd.com>: Clean up -- 2c8c71e46e10d5349b2f44f638c49a4801234c71 by Alexandros Theodoridis <atheodor@amd.com>: Clean-up -- 408262b41467d7cff1cf43e848428894440aa290 by Alexandros Theodoridis <atheodor@amd.com>: Address review comments Merging this change closes #48140 PiperOrigin-RevId: 975346419
…data-formatting ops and fusions. - Update FindConstrainedUses to recursively trace through transparent data-formatting ops (copy, reshape, transpose, slice, bitcast, get_tuple_element) and fusions using a single-pass DFS. - Add IdentityElementType support for kMinimum and kMaximum. - Add unit tests verifying min/max reduction identity element generation through copies and fusions. PiperOrigin-RevId: 975388177
PiperOrigin-RevId: 975392160
Sometime while resolving export issues I made manual changes and didn't rerun my validation steps; I've triple-checked this time. PiperOrigin-RevId: 975394784
…ation Update InfeedTokenPropagation to preserve and adjust instruction OriginalValue when tuplifying instruction shapes and inserting tokens into tuples. PiperOrigin-RevId: 975397101
… of addStablehloImportPipeline() in stablehlo_import.cc. This canonicalizes tuple-returning operations into multi-result operations before HLO sharding import. Added FileCheck unit tests to stablehlo_import_pipeline.mlir verifying HLO sharding import on tuple-returning custom calls. PiperOrigin-RevId: 975399339
…vices_for_output_map. Migrate RemapPlan callers in IFRT proxy client to directly construct `input_devices_for_output_map` without deprecated `mappings`. In `Array::RemapArrays`, use `plan.output_specs()[i].layout` directly instead of deriving output layouts from `plan.mappings()`. In `array_test.cc`, construct `RemapPlan` with `input_devices_for_output_map` instead of `mappings`. PiperOrigin-RevId: 975401333
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.
12 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 : )