Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[XLA][HloDCE] Removal of unused outputs of fusions to consider multiple users of the same shape index (aka output) #17217

Merged
merged 1 commit into from
Sep 17, 2024

Commits on Sep 17, 2024

  1. [XLA][HloDCE] Removal of unused outputs of fusions to consider multip…

    …le users of the same shape index (aka output)
    
    Changes:
    * Replaces check of users vs number of outputs with checking if number of unique outputs used is smaller than number of outputs
    * Before this change, if same shape index (aka same output) is used multiple times, we might not end up removing any of the unused.
    * Before this change, if an output has multiple users, and it is the only one used, we might not remove all the unused fusion outputs (aka will leave around <number of users> outputs).
    PiperOrigin-RevId: 675754198
    Google-ML-Automation committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    2b1c609 View commit details
    Browse the repository at this point in the history