Commit 8032d9f7 authored by Thomas Joerg's avatar Thomas Joerg Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Extract and generalize fusion-related logic from compiler...

[XLA:GPU] Extract and generalize fusion-related logic from compiler pass-specific classes: LayoutsAreReduceInputFusionFriendly

This is the first of a series of CLs. The idea is to extract and generalize IsFusable... logic scattered across GpuInstructionFusion, FusionMerger, and GpuMultiOutputFusion into the joint util. Eventually, functions like LayoutsAreReduceInputFusionFriendly will no longer be directly exposed, but rather used internally to determine fusibility of given ops.

PiperOrigin-RevId: 210888913
parent ea31ddf2
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment