fix compilation crash

binarman · binarman · commit ee86c630c86b · 2025-10-31T13:37:36.000Z
diff --git a/third_party/amd/lib/TritonAMDGPUTransforms/OptimizeBufferOpPtr.cpp b/third_party/amd/lib/TritonAMDGPUTransforms/OptimizeBufferOpPtr.cpp
@@ -157,7 +157,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {
     int offsetOperandNo = blockArg.getArgNumber() - forOp.getNumInductionVars();
     auto offsetYieldOperand = yield.getOperand(offsetOperandNo);
     auto incrementOp = offsetYieldOperand.getDefiningOp();
-    if (!isa<arith::AddIOp>(incrementOp)) {
+    if (!incrementOp || !isa<arith::AddIOp>(incrementOp)) {
       LDBG("Rejected: expect arith::addi used for pointer advanceent");
       return {};
     }

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {`
`157`	`157`	`int offsetOperandNo = blockArg.getArgNumber() - forOp.getNumInductionVars();`
`158`	`158`	`auto offsetYieldOperand = yield.getOperand(offsetOperandNo);`
`159`	`159`	`auto incrementOp = offsetYieldOperand.getDefiningOp();`
`160`		`- if (!isa<arith::AddIOp>(incrementOp)) {`
	`160`	`+ if (!incrementOp \|\| !isa<arith::AddIOp>(incrementOp)) {`
`161`	`161`	`LDBG("Rejected: expect arith::addi used for pointer advanceent");`
`162`	`162`	`return {};`
`163`	`163`	`}`