Skip to content

Commit

Permalink
[XLA:GPU][Emitters] Restrict the inliner.
Browse files Browse the repository at this point in the history 
Inline only if there are more than 1 call to the callee in the caller.

Background: jax-ml/jax#26162 contains an example of a MoF fusion that takes forever to compile.

The [indexing-based partitioner](44bc816) in combination with this change fixes the issue.

PiperOrigin-RevId: 730436982
  • Loading branch information
@pifon2a @Google-ML-Automation
pifon2a authored and Google-ML-Automation committed Feb 24, 2025
1 parent 320e114 commit 2e141f6
Show file tree
Hide file tree
Showing 3 changed files with 361 additions and 19 deletions.
Loading

0 comments on commit 2e141f6

Please sign in to comment.