-
- Downloads
[RUNTIME] Enable extension type to PackedFunc. (#447)
* [RUNTIME] Enable extension type to PackedFunc. * More comments
Showing
- apps/extension/python/tvm_ext/__init__.py 23 additions, 0 deletionsapps/extension/python/tvm_ext/__init__.py
- apps/extension/src/tvm_ext.cc 32 additions, 0 deletionsapps/extension/src/tvm_ext.cc
- apps/extension/tests/test_ext.py 13 additions, 0 deletionsapps/extension/tests/test_ext.py
- include/tvm/packed_func_ext.h 6 additions, 6 deletionsinclude/tvm/packed_func_ext.h
- include/tvm/runtime/c_runtime_api.h 20 additions, 1 deletioninclude/tvm/runtime/c_runtime_api.h
- include/tvm/runtime/packed_func.h 175 additions, 47 deletionsinclude/tvm/runtime/packed_func.h
- include/tvm/runtime/registry.h 15 additions, 2 deletionsinclude/tvm/runtime/registry.h
- python/tvm/_ffi/_ctypes/function.py 1 addition, 1 deletionpython/tvm/_ffi/_ctypes/function.py
- python/tvm/_ffi/_ctypes/ndarray.py 7 additions, 1 deletionpython/tvm/_ffi/_ctypes/ndarray.py
- python/tvm/_ffi/_cython/base.pxi 1 addition, 0 deletionspython/tvm/_ffi/_cython/base.pxi
- python/tvm/_ffi/_cython/function.pxi 8 additions, 4 deletionspython/tvm/_ffi/_cython/function.pxi
- python/tvm/_ffi/_cython/ndarray.pxi 6 additions, 1 deletionpython/tvm/_ffi/_cython/ndarray.pxi
- python/tvm/_ffi/ndarray.py 26 additions, 10 deletionspython/tvm/_ffi/ndarray.py
- python/tvm/_ffi/runtime_ctypes.py 1 addition, 0 deletionspython/tvm/_ffi/runtime_ctypes.py
- python/tvm/ndarray.py 2 additions, 1 deletionpython/tvm/ndarray.py
- src/runtime/registry.cc 33 additions, 0 deletionssrc/runtime/registry.cc
- tests/cpp/packed_func_test.cc 49 additions, 0 deletionstests/cpp/packed_func_test.cc
- tests/python/unittest/test_runtime_extension.py 1 addition, 4 deletionstests/python/unittest/test_runtime_extension.py
Loading
Please register or sign in to comment