[tune] Introduce transform.tune.select and tune dialect boilerplate (#1053)

Can be used to generate syntax which represents a non-deterministic
choice. E.g., as follows:
```python
diff --git a/python/mlir/tpp/sched/main.py b/python/mlir/tpp/sched/main.py
index be31fb2..4f1f6719 100644
--- a/python/mlir/tpp/sched/main.py
+++ b/python/mlir/tpp/sched/main.py
@@ -5,6 +5,7 @@ from argparse import ArgumentParser
 from mlir import ir, passmanager
 from mlir.dialects import transform
 from mlir.dialects import check, perf, xsmm  # Registers our TPP-MLIR dialects.
+from mlir.dialects.transform import tune
 from mlir.dialects.transform import interpreter as transform_interpreter
 
 from . import bundles
@@ -92,6 +93,12 @@ def overall_schedule(**config: Dict[str, Any]) -> ir.Module:
                 op_name="builtin.module",
                 deduplicate=True,
             )
+            i64 = ir.IntegerType.get_signless(64)
+            tune.select(
+                transform.AnyParamType.get(),
+                name="coin_flip",
+                options=[ir.IntegerAttr.get(i64, 0), ir.IntegerAttr.get(i64, 1)],
+            )
 
             try:
                 # Now pass the module handle through the bundles.
```

Author: rolfmorel

include/TPP/Dialect/CMakeLists.txt

@@ -1,3 +1,4 @@
 add_subdirectory(Check)
 add_subdirectory(Perf)
+add_subdirectory(Tune)
 add_subdirectory(Xsmm)

include/TPP/Dialect/Tune/CMakeLists.txt

@@ -0,0 +1,9 @@
+#add_mlir_dialect(TuneOps tune)
+add_mlir_doc(TuneDialect TuneDialect TPP/ -gen-dialect-doc)
+
+set(LLVM_TARGET_DEFINITIONS TuneTransformOps.td)
+mlir_tablegen(TuneTransformOps.h.inc -gen-op-decls)
+mlir_tablegen(TuneTransformOps.cpp.inc -gen-op-defs)
+add_public_tablegen_target(TuneTransformOpsIncGen)
+
+add_mlir_doc(TuneTransformOps TuneTransformOps ./ -gen-op-doc)

include/TPP/Dialect/Tune/TuneDialect.h

@@ -0,0 +1,26 @@
+//===- TuneDialect.h - Tune dialect -----------------------------*- C++ -*-===//
+//
+// This file is licensed 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
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef TPP_DIALECT_TUNE_TUNEDIALECT_H
+#define TPP_DIALECT_TUNE_TUNEDIALECT_H
+
+#include "mlir/IR/Dialect.h"
+
+namespace mlir {
+namespace tune {
+
+class TuneDialect : public Dialect {
+public:
+  explicit TuneDialect(MLIRContext *context);
+  static StringRef getDialectNamespace() { return "tune"; }
+};
+
+} // namespace tune
+} // namespace mlir
+
+#endif // TPP_DIALECT_TUNE_TUNEDIALECT_H

include/TPP/Dialect/Tune/TuneDialect.td

@@ -0,0 +1,35 @@
+//===- TuneDialect.td - Tune dialect ----------------------*- tablegen -*--===//
+//
+// This file is licensed 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
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef TPP_TUNE_DIALECT
+#define TPP_TUNE_DIALECT
+
+include "mlir/IR/OpBase.td"
+
+//===----------------------------------------------------------------------===//
+// Tune dialect definition.
+//===----------------------------------------------------------------------===//
+
+def Tune_Dialect : Dialect {
+    let name = "tune";
+    let summary = "Tune dialect.";
+    let description = [{
+        TODO
+    }];
+    let cppNamespace = "::mlir::tune";
+    let usePropertiesForAttributes = 1;
+}
+
+//===----------------------------------------------------------------------===//
+// Base operation definition.
+//===----------------------------------------------------------------------===//
+
+class Tune_Op<string mnemonic, list<Trait> traits = []> :
+        Op<Tune_Dialect, mnemonic, traits>;
+
+#endif // TPP_TUNE_DIALECT

include/TPP/Dialect/Tune/TuneTransformOps.h

@@ -0,0 +1,17 @@
+#ifndef TPP_DIALECT_TUNE_TUNETRANSFORMOPS_H
+#define TPP_DIALECT_TUNE_TUNETRANSFORMOPS_H
+
+#include "mlir/Dialect/Transform/IR/TransformOps.h"
+#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
+#include "mlir/IR/OpImplementation.h"
+
+#define GET_OP_CLASSES
+#include "TPP/Dialect/Tune/TuneTransformOps.h.inc"
+
+namespace mlir {
+namespace tune {
+void registerTransformDialectExtension(DialectRegistry &registry);
+} // namespace tune
+} // namespace mlir
+
+#endif // MLIR_TUNE_TRANSFORM_OPS_H

include/TPP/Dialect/Tune/TuneTransformOps.td

@@ -0,0 +1,25 @@
+#ifndef TUNE_TRANSFORM_OPS
+#define TUNE_TRANSFORM_OPS
+
+include "mlir/Dialect/Transform/IR/TransformDialect.td"
+include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.td"
+include "mlir/Interfaces/SideEffectInterfaces.td"
+
+
+def TuneSelectOp : Op<Transform_Dialect, "tune.select", [
+  DeclareOpInterfaceMethods<TransformOpInterface>,
+  DeclareOpInterfaceMethods<MemoryEffectsOpInterface>
+]> {
+  let summary = "Non-deterministically select a value from a set of values";
+  let description = [{
+    TODO
+  }];
+
+  let arguments = (ins SymbolRefAttr:$name,
+                       ArrayAttr:$options);
+  let results = (outs TransformParamTypeInterface:$selected);
+  let assemblyFormat =
+      "$name `from` $options attr-dict `:` type(results)";
+}
+
+#endif // TUNE_TRANSFORM_OPS

lib/TPP-CAPI/CMakeLists.txt

@@ -3,6 +3,7 @@ add_mlir_public_c_api_library(TPPCAPI
   LINK_LIBS PUBLIC
   TPPCheckDialect
   TPPPerfDialect
+  TPPTuneDialect
   TPPXsmmDialect
   TPPTransforms
   TPPPipeline

lib/TPP/Dialect/CMakeLists.txt

@@ -1,3 +1,4 @@
 add_subdirectory(Check)
 add_subdirectory(Perf)
+add_subdirectory(Tune)
 add_subdirectory(Xsmm)

lib/TPP/Dialect/Tune/CMakeLists.txt

@@ -0,0 +1,13 @@
+add_subdirectory(TransformOps)
+
+add_mlir_dialect_library(TPPTuneDialect
+  # Ops and dialects
+    TuneDialect.cpp
+
+  ADDITIONAL_HEADER_DIRS
+    ${PROJECT_SOURCE_DIR}/include/TPP
+
+  LINK_LIBS PUBLIC
+    MLIRIR
+    MLIRInferTypeOpInterface
+)

lib/TPP/Dialect/Tune/TransformOps/CMakeLists.txt

@@ -0,0 +1,13 @@
+add_mlir_dialect_library(TuneTransformOps
+  TuneTransformOps.cpp
+
+  ADDITIONAL_HEADER_DIRS
+  ${PROJECT_SOURCE_DIR}/include/TPP
+
+  DEPENDS
+  TuneTransformOpsIncGen
+
+  LINK_LIBS PUBLIC
+  MLIRIR
+  MLIRTransformDialect
+)