Skip to content

Commit

Permalink
[XLA:GPU] Remove restriction on bitcasts being a no-op with regards…
Browse files Browse the repository at this point in the history 
… to tiling

in `SoftmaxRewriterTriton`.

This restriction was necessary before we checked tiling using
`SymbolicTileAnalysis`, but is unnecessary now.

PiperOrigin-RevId: 705100500
  • Loading branch information
@bchetioui @Google-ML-Automation
bchetioui authored and Google-ML-Automation committed Dec 12, 2024
1 parent e694619 commit aac7cc3
Show file tree
Hide file tree
Showing 3 changed files with 107 additions and 755 deletions.
Loading

0 comments on commit aac7cc3

Please sign in to comment.