From 1053582f3ad5ed68815ef07fb7aed6e56c6ca5e2 Mon Sep 17 00:00:00 2001 From: Guray Ozen Date: Fri, 7 Feb 2025 11:55:00 +0100 Subject: [PATCH] [MLIR][NVVM] Add default constructor for `nvvm.barrier` [NFC] This PR adds a default constructor to `nvvm.barrier`, making it more convenient to build the OP. --- mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td index 3d378751e798f..11226dae2c3f3 100644 --- a/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td +++ b/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td @@ -463,7 +463,17 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> { } }]; let hasVerifier = 1; + let assemblyFormat = "(`id` `=` $barrierId^)? (`number_of_threads` `=` $numberOfThreads^)? attr-dict"; + + let builders = [ + OpBuilder<(ins), [{ + return build($_builder, $_state, Value{}, Value{}); + }]>, + OpBuilder<(ins "Value":$barrierId), [{ + return build($_builder, $_state, barrierId, Value{}); + }]> + ]; } def NVVM_BarrierArriveOp : NVVM_PTXBuilder_Op<"barrier.arrive">