[Mlir-commits] [mlir] [mlir][spirv] Add support for SPV_EXT_mesh_shader extension (PR #126555)

[Jakub Kuderski]

================
@@ -0,0 +1,34 @@
+//===- MeshOps.cpp - MLIR SPIR-V Mesh Ops  --------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// Defines the mesh operations in the SPIR-V dialect.
+//
+//===----------------------------------------------------------------------===//
+
+#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"
+#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
+#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
+
+#include <optional>
+
+using namespace mlir;
+
+//===----------------------------------------------------------------------===//
+// spirv.EXT.EmitMeshTasks
+//===----------------------------------------------------------------------===//
+
+LogicalResult spirv::EXTEmitMeshTasksOp::verify() {
+  if (auto payloadOp = getPayload()) {
----------------
kuhar wrote:

Also, is this guaranteed to be an op, or can this be a block argument? The variable name implies it's an op

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