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

78 lines
2.7 KiB
CMake
Raw Normal View History

[WIP][NFC]Rename files to llvm style (#35) * Change naming style for builder directory. * Change naming style for conversion folder. * Fix case sensitivity issue. * Fix missing onnx header onnx_pb.h issue. * Rename files in Conversion to llvm style. * Rename files in Dialect to llvm style. * Path fix. * Rename files in Pass to llvm style. * Rename files in Runtime to llvm style. * Rename files in Tool to llvm style. * Rename files in Transform to llvm style. * Change comments about filenames. * Fix case. * Rename interface directory to use llvm file naming convention.
2020-03-19 16:48:09 +08:00
add_subdirectory(Interface)
[NFC] Breakdown CMake Build Scripts to Smaller Parts (#39) * 1.Break down CMake scripts to smaller libraries per-directory. 2. Move some transformations and interfaces to the right folder. 3. Fix minor merge failure of the patch renaming files to use LLVM convention. * Link OMBuilder with OMONNXOps. * 1. Update the src location of generated ONNX dialect definition. 2. Link OMONNXRewrite with OMONNXOps. * Fix path to tablegen for add_onnx_mlir_dialect_doc. * Update build script for onnx_mlir_transform. * 1. Remove comment code. 2. onnx_mlir_attribute_promotion -> OMAttributePromotion. * Name tablegen generated files with LLVM convention. * Nit: reorder libraries to link against. * Nit: Link against MLIR first.
2020-03-20 22:40:51 +08:00
add_subdirectory(Dialect)
add_subdirectory(Conversion)
[WIP][NFC]Rename files to llvm style (#35) * Change naming style for builder directory. * Change naming style for conversion folder. * Fix case sensitivity issue. * Fix missing onnx header onnx_pb.h issue. * Rename files in Conversion to llvm style. * Rename files in Dialect to llvm style. * Path fix. * Rename files in Pass to llvm style. * Rename files in Runtime to llvm style. * Rename files in Tool to llvm style. * Rename files in Transform to llvm style. * Change comments about filenames. * Fix case. * Rename interface directory to use llvm file naming convention.
2020-03-19 16:48:09 +08:00
add_subdirectory(Transform)
add_subdirectory(Tool)
add_subdirectory(Builder)
add_subdirectory(Runtime)
flatten src directory structure
2019-12-23 13:13:52 +08:00
Rapid check test (#141) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. * Checkpoint, debug.py works. * Automatically generate inputs in debug.py. * Use float. * initial support for rapidcheck tests. * Convolution test case works. * Format code. * Link library with MainUtils. * Fix CMake script error. * Fast implementation of array assertion, more detailed error analysis. * More utility for DynMemRef. * Fix linking issue. * Uncomment unit test. * Refactor to separate C++/Python ExecutionSession, enable unit test. * format code. * Verbose build. * Enable PIC option for ExecusionSession. * Fix cmake error. * Build all targets. * Fix doc to build all targets. * Clean up. * Clean up, debug. * Use type alias consistently. * Move definitions to DynMemRef.cpp. * include algorithm. * pyruntime -> PyRuntime * Format code. * Free memory. * Add comments. * Copyright notice. * Improve stylistic consistency. * Add comment. * Revert irrelevant changes. * Disambiguate. * Refator test case generator out from test case implementation, implement example exhaustive test driver. * Add documentation for testing.
2020-06-08 10:18:55 +08:00
add_library(MainUtils
[NFC] Reorganize main function. (#44) * Reorganize main function. * Follow review comments. * Use new file names.
2020-03-25 01:48:54 +08:00
MainUtils.hpp
Rapid check test (#141) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. * Checkpoint, debug.py works. * Automatically generate inputs in debug.py. * Use float. * initial support for rapidcheck tests. * Convolution test case works. * Format code. * Link library with MainUtils. * Fix CMake script error. * Fast implementation of array assertion, more detailed error analysis. * More utility for DynMemRef. * Fix linking issue. * Uncomment unit test. * Refactor to separate C++/Python ExecutionSession, enable unit test. * format code. * Verbose build. * Enable PIC option for ExecusionSession. * Fix cmake error. * Build all targets. * Fix doc to build all targets. * Clean up. * Clean up, debug. * Use type alias consistently. * Move definitions to DynMemRef.cpp. * include algorithm. * pyruntime -> PyRuntime * Format code. * Free memory. * Add comments. * Copyright notice. * Improve stylistic consistency. * Add comment. * Revert irrelevant changes. * Disambiguate. * Refator test case generator out from test case implementation, implement example exhaustive test driver. * Add documentation for testing.
2020-06-08 10:18:55 +08:00
MainUtils.cpp)
target_link_libraries(MainUtils onnx)
target_include_directories(MainUtils PRIVATE ${ONNX_MLIR_SRC_ROOT})
target_include_directories(MainUtils PRIVATE ${CMAKE_BINARY_DIR})
target_include_directories(MainUtils PRIVATE ${ONNX_MLIR_BIN_ROOT})
add_executable(onnx-mlir
[NFC] Reorganize main function. (#44) * Reorganize main function. * Follow review comments. * Use new file names.
2020-03-25 01:48:54 +08:00
main.cpp)
Rapid check test (#141) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. * Checkpoint, debug.py works. * Automatically generate inputs in debug.py. * Use float. * initial support for rapidcheck tests. * Convolution test case works. * Format code. * Link library with MainUtils. * Fix CMake script error. * Fast implementation of array assertion, more detailed error analysis. * More utility for DynMemRef. * Fix linking issue. * Uncomment unit test. * Refactor to separate C++/Python ExecutionSession, enable unit test. * format code. * Verbose build. * Enable PIC option for ExecusionSession. * Fix cmake error. * Build all targets. * Fix doc to build all targets. * Clean up. * Clean up, debug. * Use type alias consistently. * Move definitions to DynMemRef.cpp. * include algorithm. * pyruntime -> PyRuntime * Format code. * Free memory. * Add comments. * Copyright notice. * Improve stylistic consistency. * Add comment. * Revert irrelevant changes. * Disambiguate. * Refator test case generator out from test case implementation, implement example exhaustive test driver. * Add documentation for testing.
2020-06-08 10:18:55 +08:00
target_link_libraries(onnx-mlir MainUtils)
Specify each lib only once; allow llvm build in shared libs mode. (#77) * Specify each lib only once; allow llvm build in shared libs mode. * Remove debug code. * For library targets, retain dependency information using add_dependencies, but do not link using taget_link_libraries. * Do not set LD_PRELOAD by default. Co-authored-by: Gong Su <gongsu@us.ibm.com> Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-04-14 17:40:05 +08:00
Call llc, ld from within onnx-mlir. (#127) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-05-19 10:15:48 +08:00
# Locate llc, which is needed for translating LLVM bitcode
# to object file.
if(NOT EXISTS "${LLVM_PROJ_BUILD}/bin/llc")
message(ERROR "Cannot find llc.")
endif()
# Get the compiler command name, the C++ compiler is needed to to translate
# object files to shared libraries.
get_filename_component(CXX_COMPILER_FILENAME ${CMAKE_CXX_COMPILER} NAME)
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/ExternalUtil.hpp.in
${CMAKE_CURRENT_BINARY_DIR}/ExternalUtil.hpp)
Specify each lib only once; allow llvm build in shared libs mode. (#77) * Specify each lib only once; allow llvm build in shared libs mode. * Remove debug code. * For library targets, retain dependency information using add_dependencies, but do not link using taget_link_libraries. * Do not set LD_PRELOAD by default. Co-authored-by: Gong Su <gongsu@us.ibm.com> Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-04-14 17:40:05 +08:00
set(ONNX_MLIR_LD_PRELOAD_onnx-mlir "" CACHE STRING "" FORCE)
whole_archive_link_mlir(onnx-mlir ${MLIRWholeArchiveLibs})
if(BUILD_SHARED_LIBS)
message(STATUS "To run dynamically linked onnx-mlir, you must specify:")
message(STATUS "LD_PRELOAD=${ONNX_MLIR_LD_PRELOAD_onnx-mlir}")
endif()
# Libraries specified on the target_link_libraries for the add_subdirectory
# targets get added to the end of the list here. This creates two problems:
# 1. It produces duplicated libraries being specified for the link command.
# 2. The libraries added at the end may depend on other libraries thus
# cause linkage errors due to undefined symbols.
# So it's better not to use target_link_libraries for the add_subdirectory
# targets, or only use it for libraries that have no further dependencies
# (except system libraries such as libc).
Rapid check test (#141) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. * Checkpoint, debug.py works. * Automatically generate inputs in debug.py. * Use float. * initial support for rapidcheck tests. * Convolution test case works. * Format code. * Link library with MainUtils. * Fix CMake script error. * Fast implementation of array assertion, more detailed error analysis. * More utility for DynMemRef. * Fix linking issue. * Uncomment unit test. * Refactor to separate C++/Python ExecutionSession, enable unit test. * format code. * Verbose build. * Enable PIC option for ExecusionSession. * Fix cmake error. * Build all targets. * Fix doc to build all targets. * Clean up. * Clean up, debug. * Use type alias consistently. * Move definitions to DynMemRef.cpp. * include algorithm. * pyruntime -> PyRuntime * Format code. * Free memory. * Add comments. * Copyright notice. * Improve stylistic consistency. * Add comment. * Revert irrelevant changes. * Disambiguate. * Refator test case generator out from test case implementation, implement example exhaustive test driver. * Add documentation for testing.
2020-06-08 10:18:55 +08:00
target_link_libraries(MainUtils
[NFC] Breakdown CMake Build Scripts to Smaller Parts (#39) * 1.Break down CMake scripts to smaller libraries per-directory. 2. Move some transformations and interfaces to the right folder. 3. Fix minor merge failure of the patch renaming files to use LLVM convention. * Link OMBuilder with OMONNXOps. * 1. Update the src location of generated ONNX dialect definition. 2. Link OMONNXRewrite with OMONNXOps. * Fix path to tablegen for add_onnx_mlir_dialect_doc. * Update build script for onnx_mlir_transform. * 1. Remove comment code. 2. onnx_mlir_attribute_promotion -> OMAttributePromotion. * Name tablegen generated files with LLVM convention. * Nit: reorder libraries to link against. * Nit: Link against MLIR first.
2020-03-20 22:40:51 +08:00
OMBuilder
OMKrnlOps
OMONNXOps
Specify each lib only once; allow llvm build in shared libs mode. (#77) * Specify each lib only once; allow llvm build in shared libs mode. * Remove debug code. * For library targets, retain dependency information using add_dependencies, but do not link using taget_link_libraries. * Do not set LD_PRELOAD by default. Co-authored-by: Gong Su <gongsu@us.ibm.com> Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-04-14 17:40:05 +08:00
OMShapeInference
OMShapeInferenceOpInterface
OMAttributePromotion
OMPromotableConstOperandsOpInterface
fix type inference (#144) * fix type inference for ConstantOp and MaxPoolSingleOut * modify interface * use OpInterface * change name to OpInterface * Builder dependence * Update CMakeLists.txt * Update CMakeLists.txt Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-05-26 09:54:19 +08:00
OMResultTypeInferenceOpInterface
Rework output to improve readability of intermediate MLIR code. (#87) * Reorganize main function. * Follow review comments. * Emit constants are globals in Krnl and LLVM dialects. * Output of non-value constants. Write full source to file. * Fix e2e tests. * Output constant free and full code in separate files. * Emit separate files. * Move file output management to utils. * Elide the values of glotbal krnl constants. * Add dual file output for Basic flag. * Add tests. * Add passes to cmake file.
2020-04-25 04:15:36 +08:00
OMElideConstants
OMElideKrnlGlobalConstants
[NFC] Breakdown CMake Build Scripts to Smaller Parts (#39) * 1.Break down CMake scripts to smaller libraries per-directory. 2. Move some transformations and interfaces to the right folder. 3. Fix minor merge failure of the patch renaming files to use LLVM convention. * Link OMBuilder with OMONNXOps. * 1. Update the src location of generated ONNX dialect definition. 2. Link OMONNXRewrite with OMONNXOps. * Fix path to tablegen for add_onnx_mlir_dialect_doc. * Update build script for onnx_mlir_transform. * 1. Remove comment code. 2. onnx_mlir_attribute_promotion -> OMAttributePromotion. * Name tablegen generated files with LLVM convention. * Nit: reorder libraries to link against. * Nit: Link against MLIR first.
2020-03-20 22:40:51 +08:00
OMKrnlToAffine
OMKrnlToLLVM
OMONNXToKrnl
OMONNXRewrite
Specify each lib only once; allow llvm build in shared libs mode. (#77) * Specify each lib only once; allow llvm build in shared libs mode. * Remove debug code. * For library targets, retain dependency information using add_dependencies, but do not link using taget_link_libraries. * Do not set LD_PRELOAD by default. Co-authored-by: Gong Su <gongsu@us.ibm.com> Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-04-14 17:40:05 +08:00
${MLIRLibs}
[NFC] Breakdown CMake Build Scripts to Smaller Parts (#39) * 1.Break down CMake scripts to smaller libraries per-directory. 2. Move some transformations and interfaces to the right folder. 3. Fix minor merge failure of the patch renaming files to use LLVM convention. * Link OMBuilder with OMONNXOps. * 1. Update the src location of generated ONNX dialect definition. 2. Link OMONNXRewrite with OMONNXOps. * Fix path to tablegen for add_onnx_mlir_dialect_doc. * Update build script for onnx_mlir_transform. * 1. Remove comment code. 2. onnx_mlir_attribute_promotion -> OMAttributePromotion. * Name tablegen generated files with LLVM convention. * Nit: reorder libraries to link against. * Nit: Link against MLIR first.
2020-03-20 22:40:51 +08:00
${CMAKE_DL_LIBS})
Fix header dep (#150) * Detect llvm-project commit change in utils/clone-mlir.sh and rebuild llvm-project for zLinux Jenkins build bot * Since many headers are generated and included indirectly through other headers, there are often missing dependencies that break parallel build. So we add header targets for KrnlOps, ONNXOps, and MLONNXOps, and add explicit dependencies for all the relevant headers. * fix copy-and-paste bug Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-06-09 10:38:32 +08:00
add_dependencies(onnx-mlir OMKrnlOpsInc OMONNXOpsInc)
[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
import onnx ml operations as a separate dialect (#97) * generate MLONNX dialect * Add MLONNX to Builder * add MLONNX dialect operation document * fix the format issues Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-05-13 06:43:44 +08:00
if (INCLUDE_ONNX_ML)
Rapid check test (#141) * Call llc, ld from within onnx-mlir. * Rename EmitLLVMBC -> EmitLib., reorder header files * Edit comment. * Checkpoint, debug.py works. * Automatically generate inputs in debug.py. * Use float. * initial support for rapidcheck tests. * Convolution test case works. * Format code. * Link library with MainUtils. * Fix CMake script error. * Fast implementation of array assertion, more detailed error analysis. * More utility for DynMemRef. * Fix linking issue. * Uncomment unit test. * Refactor to separate C++/Python ExecutionSession, enable unit test. * format code. * Verbose build. * Enable PIC option for ExecusionSession. * Fix cmake error. * Build all targets. * Fix doc to build all targets. * Clean up. * Clean up, debug. * Use type alias consistently. * Move definitions to DynMemRef.cpp. * include algorithm. * pyruntime -> PyRuntime * Format code. * Free memory. * Add comments. * Copyright notice. * Improve stylistic consistency. * Add comment. * Revert irrelevant changes. * Disambiguate. * Refator test case generator out from test case implementation, implement example exhaustive test driver. * Add documentation for testing.
2020-06-08 10:18:55 +08:00
target_link_libraries(MainUtils OMMLONNXOps)
Fix header dep (#150) * Detect llvm-project commit change in utils/clone-mlir.sh and rebuild llvm-project for zLinux Jenkins build bot * Since many headers are generated and included indirectly through other headers, there are often missing dependencies that break parallel build. So we add header targets for KrnlOps, ONNXOps, and MLONNXOps, and add explicit dependencies for all the relevant headers. * fix copy-and-paste bug Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-06-09 10:38:32 +08:00
add_dependencies(MainUtils OMMLONNXOpsInc)
import onnx ml operations as a separate dialect (#97) * generate MLONNX dialect * Add MLONNX to Builder * add MLONNX dialect operation document * fix the format issues Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-05-13 06:43:44 +08:00
endif()
Make path relative to onnx-mlir project only (#40) * Make path relative to onnx-mlir project. * Use binary root folder path for onnx-mlir.
2020-03-21 00:04:22 +08:00
target_include_directories(onnx-mlir PRIVATE ${ONNX_MLIR_SRC_ROOT})
[NFC] Change ONNF based names to ONNX-MLIR (#32) * Rename onnf to onnx-mlir. * Change workspace name.
2020-03-17 21:16:33 +08:00
target_include_directories(onnx-mlir PRIVATE ${CMAKE_BINARY_DIR})
Make path relative to onnx-mlir project only (#40) * Make path relative to onnx-mlir project. * Use binary root folder path for onnx-mlir.
2020-03-21 00:04:22 +08:00
target_include_directories(onnx-mlir PRIVATE ${ONNX_MLIR_BIN_ROOT})
[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
[NFC] Change ONNF based names to ONNX-MLIR (#32) * Rename onnf to onnx-mlir. * Change workspace name.
2020-03-17 21:16:33 +08:00
install(TARGETS onnx-mlir DESTINATION bin)