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

118 lines
4.5 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_interface.hpp
dialect/onnx/onnxop.inc
pass/onnx_combine.cpp
Rewrite ReduceL1, ReduceL2, ReduceLogSum, ReduceLogSumExp, ReduceSumSquare in the ONNX dialect (#38) * Rewrite ReduceSumSquare * Edit gen_doc.py * Revise the code * Do shape inference after canonicalization so that there is no need to implement shape inference of rewritten ops * Rewrite ReduceL2 * Add onnx_rewrite.cpp for all rewriting for ONNX ops * Rewrite ReduceL1, ReduceLogSum, ReduceLogSumExp * Edit comments * Change the use of -> to . * Checkout gen_doc.py from the master branch * Use emplace_back instead of push_back * Revise the code * Edit comments Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-01-31 19:00:39 +08:00
pass/onnx_rewrite.cpp
Add a pass to decompose ONNX operations (#9)
2020-03-04 23:53:59 +08:00
pass/onnx_decompose.cpp
Revert "Support attribute promotion." This reverts commit 955968b7504ef1fb226317b81ae644e27e4c64a4.
2020-03-17 17:41:59 +08:00
pass/passes.hpp)
flatten src directory structure
2019-12-23 13:13:52 +08:00
# 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)
Add a pass to decompose ONNX operations (#9)
2020-03-04 23:53:59 +08:00
set(LLVM_TARGET_DEFINITIONS pass/onnx_decompose.td)
onnf_tablegen(onnx_decompose.inc -gen-rewriters)
add_public_tablegen_target(gen_onnx_decompose)
add_dependencies(compiler gen_onnx_decompose)
flatten src directory structure
2019-12-23 13:13:52 +08:00
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)
Add result type inference to op definition (#87) * Add result type inference to op definition * Edit MLIR tests * Fix result type for Mul * Format comments * Return UnrankedTensorType as result type * Just for testing -split-input-file * Undo: Just for testing -split-input-file * Extract a function, get_operand_ins, that gets operand types; rewrite gen_attr_ins function * Generate custom builders * Call existing build methods * Add comments * Minor changes * Generate build methods with attributes * Add support of variadic type * Do not generate custom build methods for ops having only attributes Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-02-21 22:28:24 +08:00
set(LLVM_TARGET_DEFINITIONS pass/onnx_rewrite.td)
onnf_tablegen(onnx_rewrite.inc -gen-rewriters)
add_public_tablegen_target(gen_onnx_rewrite)
add_dependencies(compiler gen_onnx_rewrite)
flatten src directory structure
2019-12-23 13:13:52 +08:00
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")
Using Tablegen to Generate Op Documentation (#74) * Add dialect documentation. * Add a step in our CI to ensure documentation is up-to-date. * Add dialect documentation. * Fix config file mistake, using multi-line commands. * Fix a bug in DocCheck.
2020-02-11 03:18:54 +08:00
set(GEN_DOC_FILE ${CMAKE_BINARY_DIR}/docs/Dialects/onnx.md)
flatten src directory structure
2019-12-23 13:13:52 +08:00
add_public_tablegen_target(gen_onnx)
add_dependencies(compiler gen_onnx)
Using Tablegen to Generate Op Documentation (#74) * Add dialect documentation. * Add a step in our CI to ensure documentation is up-to-date. * Add dialect documentation. * Fix config file mistake, using multi-line commands. * Fix a bug in DocCheck.
2020-02-11 03:18:54 +08:00
add_onnf_dialect_doc(onnx dialect/onnx/onnx.td)
flatten src directory structure
2019-12-23 13:13:52 +08:00
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 a pass to decompose ONNX operations (#9)
2020-03-04 23:53:59 +08:00
add_library(onnf_onnx_decompose pass/onnx_decompose.cpp)
target_include_directories(onnf_onnx_decompose
PRIVATE ${ONNF_SRC_ROOT} ${ONNF_BIN_ROOT}
${ONNF_SRC_ROOT})
target_link_libraries(onnf_onnx_decompose ${MLIRLibs})
Revert "Support attribute promotion." This reverts commit 955968b7504ef1fb226317b81ae644e27e4c64a4.
2020-03-17 17:41:59 +08:00
add_dependencies(onnf_onnx_decompose gen_krnl_ops)
Add a pass to decompose ONNX operations (#9)
2020-03-04 23:53:59 +08:00
flatten src directory structure
2019-12-23 13:13:52 +08:00
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)
[NFC] Change structure of conversion folder. (#96) * Change structure of conversion folder. * Fix comments. Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-02-25 23:38:08 +08:00
add_library(onnf_lower_frontend
conversion/onnx_to_krnl/onnx_to_krnl_common.cpp
conversion/onnx_to_krnl/onnx_to_krnl_common.hpp
conversion/onnx_to_krnl/math/elementwise.cpp
conversion/onnx_to_krnl/math/gemm.cpp
conversion/onnx_to_krnl/math/matmul.cpp
conversion/onnx_to_krnl/math/reduction.cpp
conversion/onnx_to_krnl/math/softmax.cpp
conversion/onnx_to_krnl/nn/conv.cpp
conversion/onnx_to_krnl/nn/normalization.cpp
Lower MaxPoolSingleOutOp to Krnl dialect (#1) * Lower MaxPoolSingleOutOp to Krnl dialect * Edit comments * Update changes according to the new folder structure * Add MLIR tests * Support ceil_mode * Merge the first two krnl loops into one krnl loop; remove attribute checks * Dynamically allocate memory for the result if the result has unknown dimensions Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-03-05 03:27:21 +08:00
conversion/onnx_to_krnl/nn/pooling.cpp
[NFC] Change structure of conversion folder. (#96) * Change structure of conversion folder. * Fix comments. Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-02-25 23:38:08 +08:00
conversion/onnx_to_krnl/tensor/identity.cpp
conversion/onnx_to_krnl/tensor/reshape.cpp
Lower constant padding operation to KRNL dialect (#27)
2020-03-12 04:54:07 +08:00
conversion/onnx_to_krnl/tensor/padconstantvaluepad.cpp
[NFC] Change structure of conversion folder. (#96) * Change structure of conversion folder. * Fix comments. Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-02-25 23:38:08 +08:00
conversion/onnx_to_krnl/tensor/transpose.cpp
conversion/onnx_to_krnl/tensor/unsqueeze.cpp
Lower ConstantOp (#28) * Lower ConstantOp * Refactor the code * Edit error messages * Check whether attribute is sparse or dense during shape inference
2020-03-12 22:58:42 +08:00
conversion/onnx_to_krnl/tensor/constant.cpp
[NFC] Change structure of conversion folder. (#96) * Change structure of conversion folder. * Fix comments. Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-02-25 23:38:08 +08:00
conversion/onnx_to_krnl/convert_onnx_to_krnl.cpp)
flatten src directory structure
2019-12-23 13:13:52 +08:00
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
Add a pass to decompose ONNX operations (#9)
2020-03-04 23:53:59 +08:00
target_link_libraries(onnf builder ${MLIRLibs} onnf_transform onnf_onnx_decompose onnf_shape_inference onnf_lower_frontend)
fix build
2019-12-23 15:09:11 +08:00
whole_archive_link_mlir(onnf ${MLIRWholeArchiveLibs})
Properly link with ZLIB. (#40)
2020-01-22 00:08:16 +08:00
find_package(ZLIB REQUIRED)
target_link_libraries(onnf ${ZLIB_LIBRARIES})
[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)