[pull] master from tensorflow:master#536
Merged
pull[bot] merged 24 commits intonoaai:masterfrom May 23, 2025
Merged
Conversation
It is beneficial to create artifical scheduling boundaries via called computations and then inline afterwards, as schedulers only act on a single computation at a time. PiperOrigin-RevId: 762467346
…es the crash in jax B200 test, that was caused by recent triton integration. PiperOrigin-RevId: 762472690
This is a cleanup. In the following changes, I want to add move dtype+reduction_kind pairs to the kernel and the amount of templating that needs to be done in this test to support different combination of reductions is getting out of hand. PiperOrigin-RevId: 762477599
create a fixture that gets the registered plugins, asserts there's only one, and then initializes it as part of the fixture. PiperOrigin-RevId: 762488288
Keeping each loop iteration as a callable simplifies scheduling drastically. PiperOrigin-RevId: 762493037
… move it to being updated only when the graph is initialized/changed and fetch from storage in the hot path. Tidy up 'Does(Occupy/Free)Resources' by computing during initialization, getting rid of std::optional / lazy init. PiperOrigin-RevId: 762495390
Skip expanding already ignored nodes when adding nodes to the ignored set. Skip expanding nodes mapped by the same matcher when mapping the subgraph. PiperOrigin-RevId: 762496848
… repacked allocations import logic to build the buffer intervals map in a `forward` manner. Previously, we used to loop over the imported allocations, and check if a collocation of it has already been added as primary (need_allocation=true) to the full_buffer_interval_map_. Instead, we now jointly import the primary and all its colocations to full_buffer_interval_map_. PiperOrigin-RevId: 762521615
PiperOrigin-RevId: 762534581
The issue here is that the tests generate random input sizes, and then compute the output size. The problem is that if the stride of the kernel is > 1, then some input sizes not aligned to the kernel stride will result in area in the input tensor that is not actually used by the convolution result, but *is* used by the dynamic quantization range computation. PiperOrigin-RevId: 762536789
…FromProto(ShapeProto)` The old constructor is deprecated. PiperOrigin-RevId: 762537880
…ailure. PiperOrigin-RevId: 762546985
PiperOrigin-RevId: 762548187
PiperOrigin-RevId: 762549281
PiperOrigin-RevId: 762552554
PiperOrigin-RevId: 762566305
Specifically, VLOG(1) --> VLOG(3) and VLOG(2) -> VLOG(4). PiperOrigin-RevId: 762568435
PiperOrigin-RevId: 762568759
…tation and workflow files for consistency PiperOrigin-RevId: 762570935
PiperOrigin-RevId: 762572277
…ue to allow insecure credentials. PiperOrigin-RevId: 762577138
PiperOrigin-RevId: 762583938
… too big to fit uint16_t type of op_params' version PiperOrigin-RevId: 762584396
PiperOrigin-RevId: 762588691
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
13 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 : )