Skip to content

Commit de7c964

Browse files
committed
Tensor descriptor may not be directly fed from create_nd_tdesc op.
For example, if passed as a block arg or func arg.
1 parent 5bb126b commit de7c964

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mlir/lib/Conversion/XeGPUToXeVM/XeGPUToXeVM.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,9 @@ class LoadStorePrefetchNdToXeVMPattern : public OpConversionPattern<OpType> {
193193
auto tdVal = op.getTensorDesc();
194194
xegpu::CreateNdDescOp descOp =
195195
tdVal.template getDefiningOp<xegpu::CreateNdDescOp>();
196+
if (!descOp)
197+
return rewriter.notifyMatchFailure(
198+
op, "Expected tensor descriptor to be created by CreateNdDescOp.");
196199
auto mixedStrides = descOp.getMixedStrides();
197200
auto mixedOffsets = op.getMixedOffsets();
198201
auto mixedSizes = descOp.getMixedSizes();

0 commit comments

Comments
 (0)