[Mlir-commits] [mlir] [mlir][NVVM] Add nvvm.membar operation (PR #166698)

[Durgadoss R]

================
@@ -291,6 +291,24 @@ static unsigned getUnidirectionalFenceProxyID(NVVM::ProxyKind fromProxy,
   llvm_unreachable("Unsupported proxy kinds");
 }
 
+static unsigned getMembarIntrinsicID(NVVM::MemScopeKind scope) {
+  switch (scope) {
+  case NVVM::MemScopeKind::CTA: {
+    return llvm::Intrinsic::nvvm_membar_cta;
+  }
+  case NVVM::MemScopeKind::CLUSTER: {
+    return llvm::Intrinsic::nvvm_fence_sc_cluster;
+  }
+  case NVVM::MemScopeKind::GPU: {
+    return llvm::Intrinsic::nvvm_membar_gl;
+  }
+  case NVVM::MemScopeKind::SYS: {
+    return llvm::Intrinsic::nvvm_membar_sys;
+  }
+  }
+  llvm_unreachable("Unknown level for memory barrier");
----------------
durga4github wrote:

Let us say "Unknown scope for memory barrier"

https://github.com/llvm/llvm-project/pull/166698