[LANG] Introduce Scan, Bugfix Canonical (#43)
Showing
- include/tvm/ir.h 31 additions, 4 deletionsinclude/tvm/ir.h
- include/tvm/operation.h 64 additions, 0 deletionsinclude/tvm/operation.h
- python/tvm/api.py 52 additions, 3 deletionspython/tvm/api.py
- python/tvm/tensor.py 7 additions, 2 deletionspython/tvm/tensor.py
- src/api/api_lang.cc 9 additions, 0 deletionssrc/api/api_lang.cc
- src/arithmetic/canonical.cc 9 additions, 2 deletionssrc/arithmetic/canonical.cc
- src/codegen/codegen_cuda.cc 1 addition, 1 deletionsrc/codegen/codegen_cuda.cc
- src/codegen/codegen_cuda.h 1 addition, 1 deletionsrc/codegen/codegen_cuda.h
- src/lang/operation.cc 87 additions, 0 deletionssrc/lang/operation.cc
- src/pass/inject_virtual_thread.cc 8 additions, 12 deletionssrc/pass/inject_virtual_thread.cc
- src/pass/storage_flatten.cc 0 additions, 34 deletionssrc/pass/storage_flatten.cc
- src/schedule/bound.cc 111 additions, 14 deletionssrc/schedule/bound.cc
- src/schedule/graph.cc 15 additions, 7 deletionssrc/schedule/graph.cc
- src/schedule/schedule_lang.cc 2 additions, 0 deletionssrc/schedule/schedule_lang.cc
- src/schedule/schedule_ops.cc 270 additions, 20 deletionssrc/schedule/schedule_ops.cc
- tests/python/integration/test_scan.py 54 additions, 0 deletionstests/python/integration/test_scan.py
- tests/python/unittest/test_lang_tensor.py 14 additions, 0 deletionstests/python/unittest/test_lang_tensor.py
- tests/python/unittest/test_pass_simplify.py 8 additions, 2 deletionstests/python/unittest/test_pass_simplify.py
- tests/python/unittest/test_schedule_schedule_ops.py 33 additions, 15 deletionstests/python/unittest/test_schedule_schedule_ops.py
Loading
Please register or sign in to comment