llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Backend/RefJIT/CMakeLists.txt

28 lines
641 B
CMake
Raw Normal View History

Initial python plumbing to interface with the refjit backend.
2020-07-11 08:36:32 +08:00
################################################################################
# NPCOMPBackendRefJITPythonModule
################################################################################
include(NpcompPython)
set(PYBIND_SOURCES
PythonModule.cpp
)
add_library(NPCOMPBackendRefJITPythonModule
${PYBIND_SOURCES}
)
target_link_libraries(NPCOMPBackendRefJITPythonModule
Start reworking towards a shared library build. * Need to have a dag of shared library deps in order to interop across python extensions (as presented in ODM). * Introduced add_npcomp_library and friends to mirror the MLIR setup. * Adds a libNPCOMP.so shared library. * Redirects tools and extensions to link against libNPCOMP.so (instead of static libs). * Moves all libraries to lib/, all binaries to bin/ and all python extensions to python/. The invariant is that the rpaths are setup to have a one level directory structure. * Reworks the _torch_mlir extension to build like the others (still need to come up with a consolidated rule to do this instead of open coded). * Includes an upstream version bump to pick up needed changes. Sizes with dynamic linking (stripped, release, asserts enabled): libNPCOMP.so: 43M (includes much of the underlying LLVM codegen deps) libMLIR.so: 31M _npcomp.so: 1.6M (python extension) _torch_mlir.so: 670K (python extension) npcomp-capi-ir-test: 6.3K npcomp-opt: 351K npcomp-run-mlir: 461K mnist-playground: 530K Still more can be done to normalize and optimize but this gets us structurally to the starting point.
2020-10-09 09:29:59 +08:00
pybind11::module
Initial python plumbing to interface with the refjit backend.
2020-07-11 08:36:32 +08:00
MLIRExecutionEngine
MLIRTargetLLVMIR
NPCOMPPythonCommon
[RefBackend] Move runtime related code under npcomp/RefBackend/ Other than the dialect definitions (which will live in standard Dialect/ subdirectory), the goal here is to keep RefBackend-related code nested in {include/npcomp,lib,test}/RefBackend.
2020-10-08 07:11:41 +08:00
NPCOMPRefBackendJITHelpers
Initial python plumbing to interface with the refjit backend.
2020-07-11 08:36:32 +08:00
)
npcomp_python_target_compile_options(NPCOMPBackendRefJITPythonModule)
Wire through codegen and runtime dependencies. * Enables e2e test. * With what I've learned in upstream about test directory layout, I can consolidate most of the separate directories we have for these things. Will do that in a followup. * Not pleased with the LLVM global initialization depends but serviceable for now.
2020-07-11 13:45:30 +08:00
add_dependencies(NPCOMPBackendRefJITPythonModule
NPCOMPPythonResources
)