Skip to content

Commit 9b41169

Browse files
committed
fix: update to new API
1 parent 4b61728 commit 9b41169

File tree

2 files changed

+164
-146
lines changed

2 files changed

+164
-146
lines changed

src/enzyme_ad/jax/Dialect/TritonExt/Ops.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,12 @@ template <>
1515
triton_ext::TritonCallOp ReadOnlyArg<triton_ext::TritonCallOp>::create(
1616
PatternRewriter &rewriter, triton_ext::TritonCallOp launchOp,
1717
ArrayRef<Type> resTys, ArrayAttr outputAliases) const {
18-
return rewriter.create<triton_ext::TritonCallOp>(
19-
launchOp.getLoc(), resTys, launchOp.getFn(), launchOp.getGridx(),
20-
launchOp.getGridy(), launchOp.getGridz(), launchOp.getClusterx(),
21-
launchOp.getClustery(), launchOp.getClusterz(), launchOp.getInputs(),
22-
launchOp.getBackendConfigAttr(), launchOp.getOperandLayoutsAttr(),
18+
return triton_ext::TritonCallOp::create(
19+
rewriter, launchOp.getLoc(), resTys, launchOp.getFn(),
20+
launchOp.getGridx(), launchOp.getGridy(), launchOp.getGridz(),
21+
launchOp.getClusterx(), launchOp.getClustery(), launchOp.getClusterz(),
22+
launchOp.getInputs(), launchOp.getBackendConfigAttr(),
23+
launchOp.getOperandLayoutsAttr(),
2324
/*resultLayouts*/ nullptr, launchOp.getArgAttrsAttr(),
2425
launchOp.getResAttrsAttr(), outputAliases,
2526
launchOp.getXlaSideEffectFreeAttr());

0 commit comments

Comments
 (0)