Commit f6709559 authored by Bixia Zheng's avatar Bixia Zheng Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Enhance kLoop fusion unroll heuristic. 

Avoid unrolling kLoop fusion with instructions such as kReduce that are likely
translated into loops in LLVM IR and can prevent vectorization.

PiperOrigin-RevId: 229246919
parent 26f2dd93
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment