Skip to content

Commit b284a84

Browse files
authored
[MLIR][NVVM] Add default constructor for nvvm.barrier [NFC] (#126225)
This PR adds a default constructor to `nvvm.barrier`, making it more convenient to build the OP.
1 parent 1f2c36a commit b284a84

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -463,7 +463,17 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> {
463463
}
464464
}];
465465
let hasVerifier = 1;
466+
466467
let assemblyFormat = "(`id` `=` $barrierId^)? (`number_of_threads` `=` $numberOfThreads^)? attr-dict";
468+
469+
let builders = [
470+
OpBuilder<(ins), [{
471+
return build($_builder, $_state, Value{}, Value{});
472+
}]>,
473+
OpBuilder<(ins "Value":$barrierId), [{
474+
return build($_builder, $_state, barrierId, Value{});
475+
}]>
476+
];
467477
}
468478

469479
def NVVM_BarrierArriveOp : NVVM_PTXBuilder_Op<"barrier.arrive">

0 commit comments

Comments
 (0)