mathic 0.2.0

A compiler with builtin support of symbolic operations, built with LLVM/MLIR
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#include "Dialect/Symbolic/Conversions/SymbolicToArith/SymbolicToArith.h"
#include "Dialect/Symbolic/IR/SymbolicDialect.h"
#include "Dialect/Symbolic/Transforms/SymbolicExtractEval.h"
#include <mlir/InitAllDialects.h>
#include <mlir/InitAllPasses.h>
#include <mlir/Pass/PassManager.h>
#include <mlir/Pass/PassRegistry.h>
#include <mlir/Tools/mlir-opt/MlirOptMain.h>
#include <mlir/Transforms/Passes.h>

namespace
{
void symbolicExtractEvalPipeline(mlir::OpPassManager &manager)
{
    manager.addPass(mlir::symbolic::createSymbolicExtractEval());
}

void symbolicToArithPipeline(mlir::OpPassManager &manager)
{
    manager.addPass(mlir::symbolic::createSymbolicToArith());
}
} // namespace

int main(int argc, char **argv)
{
    mlir::DialectRegistry registry;

    registry.insert<mlir::symbolic::SymbolicDialect>();

    mlir::registerAllDialects(registry);

    mlir::registerAllPasses();

    mlir::PassPipelineRegistration<>("symbolic-extract-eval",
                                     "Run pass to pass to convert eval operations in to function calls",
                                     symbolicExtractEvalPipeline);
    mlir::PassPipelineRegistration<>("symbolic-to-arith", "Run pass to pass to convert symbolic dialect to arith",
                                     symbolicToArithPipeline);
    return mlir::asMainReturnCode(mlir::MlirOptMain(argc, argv, "Dialect Driver", registry));
}