Support rank-0 tensor (#687)
* Support rank-0 tensor * fix lint
Showing
- include/tvm/buffer.h 5 additions, 0 deletionsinclude/tvm/buffer.h
- include/tvm/packed_func_ext.h 4 additions, 0 deletionsinclude/tvm/packed_func_ext.h
- include/tvm/tensor.h 1 addition, 1 deletioninclude/tvm/tensor.h
- python/tvm/_ffi/ndarray.py 4 additions, 3 deletionspython/tvm/_ffi/ndarray.py
- python/tvm/api.py 0 additions, 1 deletionpython/tvm/api.py
- python/tvm/tensor.py 7 additions, 1 deletionpython/tvm/tensor.py
- src/arithmetic/compute_expr.h 9 additions, 3 deletionssrc/arithmetic/compute_expr.h
- src/lang/buffer.cc 19 additions, 18 deletionssrc/lang/buffer.cc
- src/pass/arg_binder.cc 11 additions, 9 deletionssrc/pass/arg_binder.cc
- src/pass/inject_double_buffer.cc 2 additions, 1 deletionsrc/pass/inject_double_buffer.cc
- src/pass/inject_virtual_thread.cc 2 additions, 1 deletionsrc/pass/inject_virtual_thread.cc
- src/pass/storage_flatten.cc 11 additions, 4 deletionssrc/pass/storage_flatten.cc
- src/pass/storage_rewrite.cc 1 addition, 1 deletionsrc/pass/storage_rewrite.cc
- src/runtime/c_runtime_api.cc 7 additions, 3 deletionssrc/runtime/c_runtime_api.cc
- src/runtime/graph/graph_runtime.cc 4 additions, 2 deletionssrc/runtime/graph/graph_runtime.cc
- src/schedule/schedule_dataflow_rewrite.cc 3 additions, 3 deletionssrc/schedule/schedule_dataflow_rewrite.cc
- tests/python/unittest/test_codegen_device.py 5 additions, 4 deletionstests/python/unittest/test_codegen_device.py
- tests/python/unittest/test_codegen_llvm.py 25 additions, 0 deletionstests/python/unittest/test_codegen_llvm.py
- tests/python/unittest/test_lang_tensor.py 13 additions, 0 deletionstests/python/unittest/test_lang_tensor.py
- tests/python/unittest/test_runtime_packed_func.py 8 additions, 0 deletionstests/python/unittest/test_runtime_packed_func.py
Loading
Please register or sign in to comment