73140daebb
This functionality is available in C++, make it available in Python directly to operate on transform modules.
84 lines
3.0 KiB
C++
84 lines
3.0 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/IR/Utils.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)));
|
|
}
|
|
|
|
MlirLogicalResult mlirMergeSymbolsIntoFromClone(MlirOperation target,
|
|
MlirOperation other) {
|
|
OwningOpRef<Operation *> otherOwning(unwrap(other)->clone());
|
|
LogicalResult result = transform::detail::mergeSymbolsInto(
|
|
unwrap(target), std::move(otherOwning));
|
|
return wrap(result);
|
|
}
|
|
}
|