colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/src/CMakeLists.txt

81 lines
2.8 KiB
CMake
Raw Normal View History

flatten src directory structure
2019-12-23 13:13:52 +08:00
add_library(compiler
dialect/krnl/krnl_ops.cpp
dialect/krnl/krnl_ops.hpp
dialect/krnl/krnl_types.cpp
dialect/krnl/krnl_types.hpp
dialect/onnx/onnx_ops.cpp
dialect/onnx/onnx_ops.hpp
dialect/krnl/krnl_helper.cpp
dialect/krnl/krnl_helper.hpp
pass/shape_inference_pass.cpp
pass/shape_inference_interface.hpp
dialect/onnx/onnxop.inc
pass/onnx_combine.cpp
pass/lower_frontend_to_krnl.cpp
pass/passes.hpp)
# Include root src directory.
target_include_directories(compiler PRIVATE ${ONNF_SRC_ROOT})
# Include tablegen generated header files.
target_include_directories(compiler PRIVATE ${ONNF_BIN_ROOT})
target_link_libraries(compiler
${CMAKE_THREAD_LIBS_INIT}
${CMAKE_DL_LIBS}
${MLIRLibs}
curses)
set(LLVM_TARGET_DEFINITIONS pass/shape_inference_interface.td)
onnf_tablegen(shape_inference.hpp.inc -gen-op-interface-decls)
onnf_tablegen(shape_inference.cpp.inc -gen-op-interface-defs)
add_public_tablegen_target(gen_shape_inference)
add_dependencies(compiler gen_shape_inference)
set(LLVM_TARGET_DEFINITIONS pass/onnx_combine.td)
onnf_tablegen(onnx_combine.inc -gen-rewriters)
add_public_tablegen_target(gen_onnx_combine)
add_dependencies(compiler gen_onnx_combine)
set(LLVM_TARGET_DEFINITIONS dialect/onnx/onnx.td)
onnf_tablegen(onnx.hpp.inc -gen-op-decls "-I${CMAKE_SOURCE_DIR}/compiler/pass")
onnf_tablegen(onnx.cpp.inc -gen-op-defs "-I${CMAKE_SOURCE_DIR}/compiler/pass")
add_public_tablegen_target(gen_onnx)
add_dependencies(compiler gen_onnx)
set(LLVM_TARGET_DEFINITIONS dialect/krnl/krnl_ops.td)
onnf_tablegen(krnl.hpp.inc -gen-op-decls)
onnf_tablegen(krnl.cpp.inc -gen-op-defs)
add_public_tablegen_target(gen_krnl_ops)
add_dependencies(compiler gen_krnl_ops)
add_library(onnf_shape_inference pass/shape_inference_pass.cpp)
target_include_directories(onnf_shape_inference
PRIVATE ${ONNF_SRC_ROOT} ${ONNF_BIN_ROOT}
${ONNF_SRC_ROOT})
target_link_libraries(onnf_shape_inference ${MLIRLibs})
add_dependencies(onnf_shape_inference gen_krnl_ops)
add_library(onnf_lower_frontend pass/lower_frontend_to_krnl.cpp)
target_include_directories(onnf_lower_frontend
PRIVATE ${ONNF_SRC_ROOT} ${ONNF_BIN_ROOT}
${ONNF_SRC_ROOT})
target_link_libraries(onnf_lower_frontend ${MLIRLibs})
add_dependencies(onnf_lower_frontend gen_krnl_ops)
add_subdirectory(transform)
add_subdirectory(tool)
add_subdirectory(builder)
add_subdirectory(runtime)
Enable building of first commit.
2019-12-20 03:24:37 +08:00
add_executable(onnf main.cpp)
[MLIR] Lowering of frontend dialect to KRNL dialect (#382) * Partial support for lowering operations to KRNL dialect. * Attempt to lower to KRNL IR. * Update file. * Add lowering. * Address comments. Fix alloc dynamic dimensions. Correctly link StandardOps. * Temporarily remove deallocation of locally allocated tensors.
2019-11-27 02:55:44 +08:00
fix build
2019-12-23 15:09:11 +08:00
target_link_libraries(onnf builder ${MLIRLibs} onnf_transform)
clean up, remove dependency for boost
2019-12-23 09:33:33 +08:00
set_target_properties(onnf PROPERTIES LINK_FLAGS "-lz")
fix build
2019-12-23 15:09:11 +08:00
whole_archive_link_mlir(onnf ${MLIRWholeArchiveLibs})
[MLIR] import attribute of onnx node (#383) * add attributes as NamedAttribute * support list value for attribute * use std::tie to avoid c++17 feature
2019-12-21 14:58:23 +08:00
ONNX Dialect creation and registration with MLIR (#358) * Create and register ONNX Dialect with one Add operation. * Fix file formatting. * Change name from ONNX to SGIR. * Use ONNX dialect. Change SGIR to frontend placeholder dialect. * Add comments. * Type clean-up.
2019-11-02 05:09:48 +08:00
target_include_directories(onnf PRIVATE ${CMAKE_SOURCE_DIR})
target_include_directories(onnf PRIVATE ${CMAKE_BINARY_DIR})
[MLIR] Krnl Dialect Definition (#357) * a complete, roud-trippable Krnl dialect operation definition * remove old dialect definition files, edit build files * register dialect * check in src for onnf_opt and dimension handler types * re-trigger jenkins * fix build * clarify operation semantics * add verifier for krnl.iterate * refactor to make things clear * do not hard code types * nit and add comments * fix rebase * update op implementation * fix merge * update kernel dialect definition * more comment on how to use the builder for krnl.iterate operation * ammend the comment * can parse krnl.iterate * can parse and print if bounds are not SSA values * address comments * better error handling * Update CMakeLists.txt * update comment * reflow comments
2019-11-12 10:31:56 +08:00
[MLIR] Refactor Krnl Dialect and Krnl Dialect Lowering (#375) * Store bounds as affine map attributes & check in test cases with generic printer * Upgrading MLIR MLIR is outdated on Buildbot, rebuilding a newer version. * work with new version of mlir * check-in parser tests * custom printer * nit * bug fix * enable custom asm printer test * enable custom asm printer test * more consistent variable naming * test max/min * variable naming scheme change to MLIR style * can lower krnl to llvm * kernel -> llvm * comments * bug fix * try fixing ci * fix ci * deactivate model test * fix lit test * nit * fix z buildbot
2019-11-28 11:56:34 +08:00
install(TARGETS onnf DESTINATION bin)