Skip to content

Commit 9776850

Browse files
committed
cleanup
1 parent ae22f27 commit 9776850

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/lib/Dialect/XeGPU/Transforms/XeGPUBlocking.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,20 +188,20 @@ bool XeGPUBlockingPass::needsUnroll(Operation *op) const {
188188
for (auto &opr : op->getOpOperands()) {
189189
std::optional<SmallVector<int64_t>> tileShape = getTileShape(opr);
190190
auto shapedType = dyn_cast<ShapedType>(opr.get().getType());
191-
if (!shapedType)
191+
if (!shapedType || !tileShape)
192192
continue;
193193

194-
if (tileShape && !llvm::equal(*tileShape, shapedType.getShape()))
194+
if (!llvm::equal(*tileShape, shapedType.getShape()))
195195
return true;
196196
}
197197

198198
for (auto result : op->getOpResults()) {
199199
std::optional<SmallVector<int64_t>> tileShape = getTileShape(result);
200200
auto shapedType = dyn_cast<ShapedType>(result.getType());
201-
if (!shapedType)
201+
if (!shapedType || !tileShape)
202202
continue;
203203

204-
if (tileShape && !llvm::equal(*tileShape, shapedType.getShape()))
204+
if (!llvm::equal(*tileShape, shapedType.getShape()))
205205
return true;
206206
}
207207
return false;

0 commit comments

Comments
 (0)