Implement C++ registry to back Python target.generic_func (#892)
Showing
- include/tvm/build_module.h 235 additions, 38 deletionsinclude/tvm/build_module.h
- python/tvm/target.py 194 additions, 90 deletionspython/tvm/target.py
- src/codegen/build_module.cc 365 additions, 76 deletionssrc/codegen/build_module.cc
- src/runtime/threading_backend.cc 1 addition, 0 deletionssrc/runtime/threading_backend.cc
- tests/cpp/build_module_test.cc 1 addition, 1 deletiontests/cpp/build_module_test.cc
- tests/python/unittest/test_lang_target.py 10 additions, 5 deletionstests/python/unittest/test_lang_target.py
- topi/include/topi/cuda/dense.h 9 additions, 10 deletionstopi/include/topi/cuda/dense.h
- topi/include/topi/cuda/extern.h 1 addition, 1 deletiontopi/include/topi/cuda/extern.h
- topi/include/topi/cuda/injective.h 1 addition, 1 deletiontopi/include/topi/cuda/injective.h
- topi/include/topi/cuda/pooling.h 1 addition, 1 deletiontopi/include/topi/cuda/pooling.h
- topi/include/topi/cuda/reduction.h 2 additions, 2 deletionstopi/include/topi/cuda/reduction.h
- topi/include/topi/nn/dense.h 6 additions, 7 deletionstopi/include/topi/nn/dense.h
- topi/include/topi/rocm/dense.h 9 additions, 10 deletionstopi/include/topi/rocm/dense.h
- topi/python/topi/__init__.py 4 additions, 1 deletiontopi/python/topi/__init__.py
- topi/python/topi/generic/injective.py 1 addition, 1 deletiontopi/python/topi/generic/injective.py
- topi/python/topi/generic/nn.py 7 additions, 7 deletionstopi/python/topi/generic/nn.py
- topi/python/topi/nn/dense.py 1 addition, 1 deletiontopi/python/topi/nn/dense.py
- topi/src/topi.cc 106 additions, 27 deletionstopi/src/topi.cc
Loading
Please register or sign in to comment