From 14b3ec867f5a102ce063057ec858c2456f420662 Mon Sep 17 00:00:00 2001 From: Babar Khan <46613555+BabarZKhan@users.noreply.github.com> Date: Thu, 6 Nov 2025 11:01:05 +0100 Subject: [PATCH] [FixBug] TritonGPU: newExtOrBroadcastType assignment --- lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp b/lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp index 967d34c8f..42216c92d 100644 --- a/lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp +++ b/lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp @@ -1210,10 +1210,10 @@ void LayoutRematerialization::hoistConvertOnTopOfExtOrBroadcast( newExtOrBroadcast->setOperand(0, newConvertOp.getResult()); auto oldExtOrBroadcastType = cast(extOrBroadcatOp->getResult(0).getType()); - Type newExtOrBroadcasrType = RankedTensorType::get( + Type newExtOrBroadcastType = RankedTensorType::get( oldExtOrBroadcastType.getShape(), oldExtOrBroadcastType.getElementType(), dstEncoding); - newExtOrBroadcast->getResult(0).setType(newExtOrBroadcasrType); + newExtOrBroadcast->getResult(0).setType(newExtOrBroadcastType); IRMapping mapping; mapping.map(extOrBroadcatOp->getResult(0), newExtOrBroadcast->getResult(0)); slice.remove(extOrBroadcatOp->getResult(0));