[BUILD/CODEGEN] Allow combine multiple functions in build stage. (#169)
* [BUILD/CODEGEN] Allow combine multiple functions in build stage. * Enhance code module * fix compile
Showing
- include/tvm/lowered_func.h 13 additions, 0 deletionsinclude/tvm/lowered_func.h
- include/tvm/runtime/c_runtime_api.h 7 additions, 4 deletionsinclude/tvm/runtime/c_runtime_api.h
- python/tvm/_ffi/function.py 1 addition, 1 deletionpython/tvm/_ffi/function.py
- python/tvm/build.py 44 additions, 17 deletionspython/tvm/build.py
- python/tvm/collections.py 3 additions, 1 deletionpython/tvm/collections.py
- src/codegen/llvm/codegen_llvm.cc 8 additions, 17 deletionssrc/codegen/llvm/codegen_llvm.cc
- src/codegen/llvm/llvm_module.cc 9 additions, 2 deletionssrc/codegen/llvm/llvm_module.cc
- src/codegen/source_module.cc 5 additions, 0 deletionssrc/codegen/source_module.cc
- src/pass/lower_thread_allreduce.cc 1 addition, 0 deletionssrc/pass/lower_thread_allreduce.cc
- src/pass/split_host_device.cc 3 additions, 0 deletionssrc/pass/split_host_device.cc
- src/pass/storage_sync.cc 1 addition, 0 deletionssrc/pass/storage_sync.cc
- src/runtime/cuda/cuda_module.cc 10 additions, 4 deletionssrc/runtime/cuda/cuda_module.cc
- src/runtime/dso_module.cc 13 additions, 2 deletionssrc/runtime/dso_module.cc
- tests/python/unittest/test_codegen_llvm.py 35 additions, 0 deletionstests/python/unittest/test_codegen_llvm.py
- tests/python/unittest/test_module_load.py 38 additions, 0 deletionstests/python/unittest/test_module_load.py
Loading
Please register or sign in to comment