-
- Downloads
[ADDON] Allow piggy back nvcc compiler and code (#35)
Showing
- include/tvm/runtime/c_runtime_api.h 12 additions, 2 deletionsinclude/tvm/runtime/c_runtime_api.h
- include/tvm/runtime/packed_func.h 14 additions, 3 deletionsinclude/tvm/runtime/packed_func.h
- python/tvm/_ctypes/_function.py 11 additions, 2 deletionspython/tvm/_ctypes/_function.py
- python/tvm/_ctypes/_types.py 25 additions, 5 deletionspython/tvm/_ctypes/_types.py
- python/tvm/addon/__init__.py 1 addition, 0 deletionspython/tvm/addon/__init__.py
- python/tvm/addon/nvcc_compiler.py 55 additions, 0 deletionspython/tvm/addon/nvcc_compiler.py
- src/arithmetic/canonical.cc 9 additions, 4 deletionssrc/arithmetic/canonical.cc
- src/codegen/codegen_cuda.cc 13 additions, 1 deletionsrc/codegen/codegen_cuda.cc
- src/runtime/packed_func_registry.cc 6 additions, 0 deletionssrc/runtime/packed_func_registry.cc
- tests/python/integration/test_gemm.py 16 additions, 6 deletionstests/python/integration/test_gemm.py
- tests/python/unittest/test_runtime_packed_func.py 9 additions, 1 deletiontests/python/unittest/test_runtime_packed_func.py
Loading
Please register or sign in to comment