[pull] master from tensorflow:master#1625
Merged
Merged
Conversation
PiperOrigin-RevId: 912151397
1) Seems like BacktrackToBase() is very expensive because GetIntegerAsS64 is expensive. Cache results to avoid perf issue 2) Remove some VLOG(0) printing that was there since forever. 3) Reuse some executions of BacktrackToBase() for offsets we know are exactly the same. PiperOrigin-RevId: 912177603
Imported from GitHub PR openxla/xla#42188 Small usability improvement for collective memory requests, add an API that takes `absl::Span` of buffer addresses or allocation slices. Also collapse 3 identical structs into one. Copybara import of the project: -- 0d5546a61d0c64e497260c5453996302ada2cd4a by Eugene Zhulenev <ezhulenev@openxla.org>: [xla:gpu] Add an API to request multiple symm addrs/slices Merging this change closes #42188 PiperOrigin-RevId: 912186299
Updates LLVM usage to match [87d42c13cd6b](llvm/llvm-project@87d42c13cd6b) PiperOrigin-RevId: 912187043
transfer_dependency_events to make it easier to c-api-ify this. PiperOrigin-RevId: 912191487
Split out subcomputation HLO for independent compilation, part 3 of 3. At the end of this series, XLA:CPU and GPU will use the xla_metadata `compilation_unit="<identifier>"` tag to determine that a subcomputation should be considered a separate compilation unit for the purposes of HLO compilation. This should result in significant compilation time speedups for complex models at the cost of some potentially lost interprocedural optimization. The tag to control this may change as we iterate on the name and semantics. PiperOrigin-RevId: 912207896
…buffers PiperOrigin-RevId: 912221088
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.
5 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 : )