[Mlir-commits] [mlir] [MLIR][NVVM] Add default constructor for `nvvm.barrier` [NFC] (PR #126225)
Guray Ozen
llvmlistbot at llvm.org
Fri Feb 7 02:55:58 PST 2025
https://github.com/grypp created https://github.com/llvm/llvm-project/pull/126225
This PR adds a default constructor to `nvvm.barrier`, making it more convenient to build the OP.
>From 1053582f3ad5ed68815ef07fb7aed6e56c6ca5e2 Mon Sep 17 00:00:00 2001
From: Guray Ozen <gozen at nvidia.com>
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 3d378751e798f96..11226dae2c3f375 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">
More information about the Mlir-commits
mailing list