[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