diff --git a/lib/Dialect/TritonGPU/Transforms/Coalesce.cpp b/lib/Dialect/TritonGPU/Transforms/Coalesce.cpp index d00c00c2d..019a91d01 100644 --- a/lib/Dialect/TritonGPU/Transforms/Coalesce.cpp +++ b/lib/Dialect/TritonGPU/Transforms/Coalesce.cpp @@ -63,9 +63,13 @@ struct CoalescePass : public TritonGPUCoalesceBase { auto convertType = getTypeConverter(axisInfo, ptr, numWarps); // convert operands SmallVector newArgs; - for (auto v : op->getOperands()) - newArgs.push_back(builder.create( - op->getLoc(), convertType(v.getType()), v)); + for (auto v : op->getOperands()) { + if (v.getType().isa()) + newArgs.push_back(builder.create( + op->getLoc(), convertType(v.getType()), v)); + else + newArgs.push_back(v); + } // convert output types SmallVector newTypes; for (auto t : op->getResultTypes()) {