Fix for b/405045790. --- a/lib/Dialect/TritonGPU/Transforms/AccelerateMatmul.cpp +++ b/lib/Dialect/TritonGPU/Transforms/AccelerateMatmul.cpp @@ -361,9 +361,12 @@ static bool bwdFilter(Operation *op) { return false; } + // b/405045790: We don't want to propagate through the BroadcastOp because we + // probably don't care about the load before a broadcast as it would likely be + // small. This is just a heuristic to avoid a regression. return (op->hasTrait() && isMemoryEffectFree(op)) || isView(op) || - isa(op); }