[RUNTIME] Enable ext_dev type for quick plugin of device (#542)
* [RUNTIME] Enable ext_dev type for quick plugin of device * [TEST] Update testcase to cover all computation
Showing
- apps/extension/src/tvm_ext.cc 5 additions, 0 deletionsapps/extension/src/tvm_ext.cc
- apps/extension/tests/test_ext.py 20 additions, 0 deletionsapps/extension/tests/test_ext.py
- include/tvm/runtime/c_runtime_api.h 3 additions, 0 deletionsinclude/tvm/runtime/c_runtime_api.h
- python/tvm/__init__.py 1 addition, 1 deletionpython/tvm/__init__.py
- python/tvm/_ffi/runtime_ctypes.py 4 additions, 2 deletionspython/tvm/_ffi/runtime_ctypes.py
- python/tvm/build_module.py 1 addition, 1 deletionpython/tvm/build_module.py
- python/tvm/contrib/rpc.py 4 additions, 0 deletionspython/tvm/contrib/rpc.py
- python/tvm/ndarray.py 21 additions, 0 deletionspython/tvm/ndarray.py
- src/runtime/c_runtime_api.cc 1 addition, 0 deletionssrc/runtime/c_runtime_api.cc
- src/runtime/rocm/rocm_device_api.cc 1 addition, 1 deletionsrc/runtime/rocm/rocm_device_api.cc
- tests/python/unittest/test_codegen_device.py 10 additions, 10 deletionstests/python/unittest/test_codegen_device.py
Loading
Please register or sign in to comment