5a9bdd85ee
Transform interfaces are implemented, direction or via extensions, in libraries belonging to multiple other dialects. Those dialects don't need to depend on the non-interface part of the transform dialect, which includes the growing number of ops and transitive dependency footprint. Split out the interfaces into a separate library. This in turn requires flipping the dependency from the interface on the dialect that has crept in because both co-existed in one library. The interface shouldn't depend on the transform dialect either. As a consequence of splitting, the capability of the interpreter to automatically walk the payload IR to identify payload ops of a certain kind based on the type used for the entry point symbol argument is disabled. This is a good move by itself as it simplifies the interpreter logic. This functionality can be trivially replaced by a `transform.structured.match` operation.
75 lines
2.7 KiB
C++
75 lines
2.7 KiB
C++
//===- TransformTransforms.cpp - C Interface for Transform dialect --------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// C interface to transforms for the transform dialect.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir-c/Dialect/Transform/Interpreter.h"
|
|
#include "mlir-c/Support.h"
|
|
#include "mlir/CAPI/IR.h"
|
|
#include "mlir/CAPI/Support.h"
|
|
#include "mlir/CAPI/Wrap.h"
|
|
#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
|
|
#include "mlir/Dialect/Transform/Transforms/TransformInterpreterUtils.h"
|
|
|
|
using namespace mlir;
|
|
|
|
DEFINE_C_API_PTR_METHODS(MlirTransformOptions, transform::TransformOptions)
|
|
|
|
extern "C" {
|
|
|
|
MlirTransformOptions mlirTransformOptionsCreate() {
|
|
return wrap(new transform::TransformOptions);
|
|
}
|
|
|
|
void mlirTransformOptionsEnableExpensiveChecks(
|
|
MlirTransformOptions transformOptions, bool enable) {
|
|
unwrap(transformOptions)->enableExpensiveChecks(enable);
|
|
}
|
|
|
|
bool mlirTransformOptionsGetExpensiveChecksEnabled(
|
|
MlirTransformOptions transformOptions) {
|
|
return unwrap(transformOptions)->getExpensiveChecksEnabled();
|
|
}
|
|
|
|
void mlirTransformOptionsEnforceSingleTopLevelTransformOp(
|
|
MlirTransformOptions transformOptions, bool enable) {
|
|
unwrap(transformOptions)->enableEnforceSingleToplevelTransformOp(enable);
|
|
}
|
|
|
|
bool mlirTransformOptionsGetEnforceSingleTopLevelTransformOp(
|
|
MlirTransformOptions transformOptions) {
|
|
return unwrap(transformOptions)->getEnforceSingleToplevelTransformOp();
|
|
}
|
|
|
|
void mlirTransformOptionsDestroy(MlirTransformOptions transformOptions) {
|
|
delete unwrap(transformOptions);
|
|
}
|
|
|
|
MlirLogicalResult mlirTransformApplyNamedSequence(
|
|
MlirOperation payload, MlirOperation transformRoot,
|
|
MlirOperation transformModule, MlirTransformOptions transformOptions) {
|
|
Operation *transformRootOp = unwrap(transformRoot);
|
|
Operation *transformModuleOp = unwrap(transformModule);
|
|
if (!isa<transform::TransformOpInterface>(transformRootOp)) {
|
|
transformRootOp->emitError()
|
|
<< "must implement TransformOpInterface to be used as transform root";
|
|
return mlirLogicalResultFailure();
|
|
}
|
|
if (!isa<ModuleOp>(transformModuleOp)) {
|
|
transformModuleOp->emitError()
|
|
<< "must be a " << ModuleOp::getOperationName();
|
|
return mlirLogicalResultFailure();
|
|
}
|
|
return wrap(transform::applyTransformNamedSequence(
|
|
unwrap(payload), unwrap(transformRoot),
|
|
cast<ModuleOp>(unwrap(transformModule)), *unwrap(transformOptions)));
|
|
}
|
|
}
|