5a5c3176ef
Previously, the MLIR's python binding `smt.export_smtlib(...)` always emit `(reset)` to the end of smtlib string as a solver terminator. This PR added an option to suppress this trailing, as downstream users like python z3 module don't need it.
41 lines
1.5 KiB
C++
41 lines
1.5 KiB
C++
//===- ExportSMTLIB.cpp - C Interface to ExportSMTLIB ---------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Implements a C Interface for export SMTLIB.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir-c/Target/ExportSMTLIB.h"
|
|
#include "mlir/CAPI/IR.h"
|
|
#include "mlir/CAPI/Support.h"
|
|
#include "mlir/CAPI/Utils.h"
|
|
#include "mlir/Target/SMTLIB/ExportSMTLIB.h"
|
|
|
|
using namespace mlir;
|
|
|
|
MlirLogicalResult mlirTranslateOperationToSMTLIB(
|
|
MlirOperation module, MlirStringCallback callback, void *userData,
|
|
bool inlineSingleUseValues, bool indentLetBody, bool emitReset) {
|
|
mlir::detail::CallbackOstream stream(callback, userData);
|
|
smt::SMTEmissionOptions options;
|
|
options.inlineSingleUseValues = inlineSingleUseValues;
|
|
options.indentLetBody = indentLetBody;
|
|
options.emitReset = emitReset;
|
|
return wrap(smt::exportSMTLIB(unwrap(module), stream, options));
|
|
}
|
|
|
|
MlirLogicalResult
|
|
mlirTranslateModuleToSMTLIB(MlirModule module, MlirStringCallback callback,
|
|
void *userData, bool inlineSingleUseValues,
|
|
bool indentLetBody, bool emitReset) {
|
|
return mlirTranslateOperationToSMTLIB(
|
|
mlirModuleGetOperation(module), callback, userData, inlineSingleUseValues,
|
|
indentLetBody, emitReset);
|
|
}
|