set(MGB_MLIR_TABLEGEN_INC_BASE ${CMAKE_CURRENT_BINARY_DIR}/include/)
file(MAKE_DIRECTORY ${MGB_MLIR_TABLEGEN_INC_BASE}/megbrain/jit/mlir/ir/)
list(APPEND MGB_MLIR_TABLEGEN_INC ${MGB_MLIR_TABLEGEN_INC_BASE})

external_tablegen_library(
    NAME
        mlir_shape_inference
    TBLGEN
        MLIR
    SRCS
        "shape_inference_interface.td"
    INCLUDES
        ${MGB_MLIR_TABLEGEN_INC} ${MLIR_LLVM_INCLUDE_DIR}
    OUTS
        -gen-op-interface-decls include/megbrain/jit/mlir/ir/shape_inference_interface.h.inc
        -gen-op-interface-defs include/megbrain/jit/mlir/ir/shape_inference_interface.cpp.inc
)

external_tablegen_library(
    NAME
        mlir_op_def
    TBLGEN
        MLIR
    SRCS
        "ops.td"
    INCLUDES
        ${MGB_MLIR_TABLEGEN_INC} ${MLIR_LLVM_INCLUDE_DIR}
    OUTS
        -gen-op-decls  include/megbrain/jit/mlir/ir/ops.h.inc
        -gen-op-defs   include/megbrain/jit/mlir/ir/ops.cpp.inc
)