[XLA:GPU] Create prototype class for tiling propagation using symbolic tiles. #63880
Conversation
This file contains 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
![copybara-service[bot]](https://avatars.githubusercontent.com/in/44061?s=60&v=4){kind=small}
copybara-service
bot
force-pushed
the
test_615754693
branch
2 times, most recently
from
ed3eb7a
to
8dd2684
Compare
…c tiles. SymbolicTileAnalysis allows computing symbolic tiles for all the instructions in a given computation. All the symbolic tiles contained in an instance of SymbolicTileAnalysis are parameterized by the chosen output tile (i.e. the tile at the root instruction). A given instruction may be tiled in several different ways in order to produce the same output tile. In order to make sure all relevant tilings of an instruction are captured, SymbolicTileAnalysis stores a symbolic tile for each path from the root instruction of the computation to the relevant instruction. The class also allows evaluating the value of the concrete tiles for each (instruction, path) pair once the offset, size, and stride parameters of the output tile have been discovered. PiperOrigin-RevId: 617124497
copybara-service
bot
force-pushed
the
test_615754693
branch
from
8dd2684
to
e03741a
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
[XLA:GPU] Create prototype class for tiling propagation using symbolic tiles.
SymbolicTileAnalysis allows computing symbolic tiles for all the instructions
in a given computation. All the symbolic tiles contained in an instance of
SymbolicTileAnalysis are parameterized by the chosen output tile (i.e. the
tile at the root instruction).
A given instruction may be tiled in several different ways in order to produce
the same output tile. In order to make sure all relevant tilings of an
instruction are captured, SymbolicTileAnalysis stores a symbolic tile for
each path from the root instruction of the computation to the relevant
instruction.
The class also allows evaluating the value of the concrete tiles for each
(instruction, path) pair once the offset, size, and stride parameters of the
output tile have been discovered.