File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
mlir/lib/Dialect/XeGPU/Transforms Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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 ;
You can’t perform that action at this time.
0 commit comments