Skip to content

Commit b0e96ab

Browse files
committed
rename arguments
1 parent b89266e commit b0e96ab

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4062,8 +4062,8 @@ def NVVM_ClusterLaunchControlTryCancelOp
40624062
}];
40634063

40644064
let arguments = (ins UnitAttr:$multicast,
4065-
LLVM_PointerShared: $addr,
4066-
LLVM_PointerShared: $mbar);
4065+
LLVM_PointerShared: $smemAddress,
4066+
LLVM_PointerShared: $mbarrier);
40674067

40684068
let assemblyFormat = "(`multicast` $multicast^ `,`)? $addr `,` $mbar attr-dict";
40694069

mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2110,8 +2110,8 @@ NVVM::IDArgPair ClusterLaunchControlTryCancelOp::getIntrinsicIDAndArgs(
21102110
Operation &op, LLVM::ModuleTranslation &mt, llvm::IRBuilderBase &builder) {
21112111
auto curOp = cast<NVVM::ClusterLaunchControlTryCancelOp>(op);
21122112
llvm::SmallVector<llvm::Value *> args;
2113-
args.push_back(mt.lookupValue(curOp.getAddr()));
2114-
args.push_back(mt.lookupValue(curOp.getMbar()));
2113+
args.push_back(mt.lookupValue(curOp.getSmemAddress()));
2114+
args.push_back(mt.lookupValue(curOp.getMbarrrier()));
21152115

21162116
return curOp.getMulticast()
21172117
? NVVM::IDArgPair(

0 commit comments

Comments
 (0)