We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d5dc49 commit c3fdbbaCopy full SHA for c3fdbba
third_party/intel/lib/Analysis/AxisInfo.cpp
@@ -1010,8 +1010,12 @@ class MakeTensorPtrOpAxisInfoVisitor final
1010
getAxisInfo(triton::MakeTensorPtrOp op,
1011
ArrayRef<const dataflow::Lattice<AxisInfo> *> operands) override {
1012
LDBG("MakeTensorPtrOpAxisInfoVisitor: " << *op);
1013
- assert(op.getShape().size() == 2 && operands.size() == 7 &&
1014
- "MakeTensorPtrOp should have 2D shape");
+
+ // TODO: Extend to higher dimension tensor pointers.
1015
+ if (op.getShape().size() != 2)
1016
+ return {};
1017
1018
+ assert(operands.size() == 7 && "MakeTensorPtrOp should have 2D shape");
1019
1020
AxisInfo ptrInfo = operands[0]->getValue();
1021
AxisInfo shapeInfo0 = operands[1]->getValue();
0 commit comments