-
- Downloads
[RUNTIME] v2: runtime support for rocm (#386)
* v2: runtime support for rocm * fixed coding space errors * removed kROCM from c_runtime_api.h
Showing
- Makefile 3 additions, 3 deletionsMakefile
- include/tvm/runtime/c_runtime_api.h 1 addition, 1 deletioninclude/tvm/runtime/c_runtime_api.h
- make/config.mk 3 additions, 0 deletionsmake/config.mk
- 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/ndarray.py 15 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_common.h 51 additions, 0 deletionssrc/runtime/rocm/rocm_common.h
- src/runtime/rocm/rocm_device_api.cc 151 additions, 0 deletionssrc/runtime/rocm/rocm_device_api.cc
- src/runtime/rocm/rocm_module.cc 214 additions, 0 deletionssrc/runtime/rocm/rocm_module.cc
- src/runtime/rocm/rocm_module.h 37 additions, 0 deletionssrc/runtime/rocm/rocm_module.h
- tests/python/unittest/test_runtime_ndarray.py 1 addition, 0 deletionstests/python/unittest/test_runtime_ndarray.py
Loading
Please register or sign in to comment