CMakeLists.txt 334 字节
Newer Older
1 2 3 4
core_gather_headers()

gather_srcs(infrt_src SRCS
    trt_ops.cc
5
    trt_op_converter_pass.cc
6 7
    trt_op_teller_pass.cc
    trt_graph_fuse_pass.cc
8
    trt_graph_split_pass.cc
9 10
    )
mlir_tablegen_on(trt_ops)
11
mlir_add_rewriter(pd_lower_to_trt)
12 13 14

add_executable(trt-exec trt_exec.cc)
target_link_libraries(trt-exec infrt ${MLIR_IR_LIBS})