717ad160cd
Some checks failed
Validate Operations / validate-operations (push) Failing after 18m36s
docs)
14 lines
430 B
C++
14 lines
430 B
C++
#pragma once
|
|
|
|
#include "mlir/Dialect/Func/IR/FuncOps.h"
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
|
|
namespace onnx_mlir {
|
|
|
|
/// Resolves the function the PIM pipeline should treat as its entry point.
|
|
/// Prefers ONNX entry-point metadata, then `main_graph`, then the only
|
|
/// non-external function if the module is otherwise unambiguous.
|
|
llvm::FailureOr<mlir::func::FuncOp> getPimEntryFunc(mlir::ModuleOp moduleOp);
|
|
|
|
} // namespace onnx_mlir
|