[mlir][transform][python] Add AnyValueType to bindings.

[ingomueller-net]

Summary

This patch adds the MLIR C bindings and the corresponding Python bindings of the AnyValueType of the transform dialect.

[ingomueller-net]

• Rebasing onto current main to fix CI.

[springerm]

mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp

988–989

why has this changed?

[ingomueller-net]

• Removing spurious change introduced by accident. (Thanks @springerm!)

mlir/include/mlir-c/Dialect/Transform.h

 // AnyOpType
 //===---------------------------------------------------------------------===//
 
 MLIR_CAPI_EXPORTED bool mlirTypeIsATransformAnyOpType(MlirType type);
 
 MLIR_CAPI_EXPORTED MlirType mlirTransformAnyOpTypeGet(MlirContext ctx);
 
 //===---------------------------------------------------------------------===//
+// AnyValueType
+//===---------------------------------------------------------------------===//
+
+MLIR_CAPI_EXPORTED bool mlirTypeIsATransformAnyValueType(MlirType type);
+
+MLIR_CAPI_EXPORTED MlirType mlirTransformAnyValueTypeGet(MlirContext ctx);
+
+//===---------------------------------------------------------------------===//
 // OperationType
 //===---------------------------------------------------------------------===//
 
 MLIR_CAPI_EXPORTED bool mlirTypeIsATransformOperationType(MlirType type);
 
 MLIR_CAPI_EXPORTED MlirTypeID mlirTransformOperationTypeGetTypeID(void);
 
 MLIR_CAPI_EXPORTED MlirType

mlir/lib/Bindings/Python/DialectTransform.cpp

anyOpType.def_classmethod(
       "get",
       [](py::object cls, MlirContext ctx) {
         return cls(mlirTransformAnyOpTypeGet(ctx));
       },
       "Get an instance of AnyOpType in the given context.", py::arg("cls"),
       py::arg("context") = py::none());
 
   //===-------------------------------------------------------------------===//
+  // AnyValueType
+  //===-------------------------------------------------------------------===//
+
+  auto anyValueType =
+      mlir_type_subclass(m, "AnyValueType", mlirTypeIsATransformAnyValueType);
+  anyValueType.def_classmethod(
+      "get",
+      [](py::object cls, MlirContext ctx) {
+        return cls(mlirTransformAnyValueTypeGet(ctx));
+      },
+      "Get an instance of AnyValueType in the given context.", py::arg("cls"),
+      py::arg("context") = py::none());
+
+  //===-------------------------------------------------------------------===//
   // OperationType
   //===-------------------------------------------------------------------===//
 
   auto operationType =
       mlir_type_subclass(m, "OperationType", mlirTypeIsATransformOperationType,
                          mlirTransformOperationTypeGetTypeID);
   operationType.def_classmethod(
       "get",

mlir/lib/CAPI/Dialect/Transform.cpp

bool mlirTypeIsATransformAnyOpType(MlirType type) {
   return isa<transform::AnyOpType>(unwrap(type));
 }
 
 MlirType mlirTransformAnyOpTypeGet(MlirContext ctx) {
   return wrap(transform::AnyOpType::get(unwrap(ctx)));
 }
 
 //===---------------------------------------------------------------------===//
+// AnyValueType
+//===---------------------------------------------------------------------===//
+
+bool mlirTypeIsATransformAnyValueType(MlirType type) {
+  return isa<transform::AnyValueType>(unwrap(type));
+}
+
+MlirType mlirTransformAnyValueTypeGet(MlirContext ctx) {
+  return wrap(transform::AnyValueType::get(unwrap(ctx)));
+}
+
+//===---------------------------------------------------------------------===//
 // OperationType
 //===---------------------------------------------------------------------===//
 
 bool mlirTypeIsATransformOperationType(MlirType type) {
   return isa<transform::OperationType>(unwrap(type));
 }
 
 MlirTypeID mlirTransformOperationTypeGetTypeID(void) {

mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp

LLVM_DEBUG(
         llvm::dbgs() << "\n"; 
         llvm::interleaveComma(
           availableMappingSizes, DBGS() << "----availableMappingSizes: ");
         llvm::dbgs() << "\n";
         llvm::interleaveComma(activeIdOps, DBGS() << "----activeIdOps: ");
         llvm::dbgs() << "\n");
     // clang-format on
     for (auto [activeId, activeMappingSize, availableMappingSize] :
-         llvm::zip_equal(activeIdOps, activeMappingSizes,
-                         availableMappingSizes)) {
+         llvm::zip(activeIdOps, activeMappingSizes, availableMappingSizes)) {
       if (activeMappingSize > availableMappingSize) {
         return definiteFailureHelper(
             transformOp, forallOp,
             "Trying to map to fewer GPU threads than loop iterations but "
             "overprovisioning is not yet supported. "
             "Try additional tiling of the before mapping or map to more "
             "threads.");
       }