[SCHEDULE] Mutate dataflow in schedule, refactor Stage (#44)
Showing
- include/tvm/operation.h 11 additions, 11 deletionsinclude/tvm/operation.h
- include/tvm/schedule.h 62 additions, 8 deletionsinclude/tvm/schedule.h
- python/tvm/build.py 0 additions, 1 deletionpython/tvm/build.py
- python/tvm/schedule.py 60 additions, 0 deletionspython/tvm/schedule.py
- src/api/api_lang.cc 19 additions, 1 deletionsrc/api/api_lang.cc
- src/lang/operation.cc 3 additions, 3 deletionssrc/lang/operation.cc
- src/schedule/auto_inline_elem_wise.cc 6 additions, 19 deletionssrc/schedule/auto_inline_elem_wise.cc
- src/schedule/bound.cc 9 additions, 3 deletionssrc/schedule/bound.cc
- src/schedule/schedule_lang.cc 249 additions, 33 deletionssrc/schedule/schedule_lang.cc
- src/schedule/schedule_ops.cc 112 additions, 59 deletionssrc/schedule/schedule_ops.cc
- tests/cpp/tensor_test.cc 6 additions, 6 deletionstests/cpp/tensor_test.cc
- tests/python/integration/test_gemm.py 7 additions, 22 deletionstests/python/integration/test_gemm.py
- tests/python/unittest/test_lang_schedule.py 2 additions, 2 deletionstests/python/unittest/test_lang_schedule.py
- tests/python/unittest/test_schedule_schedule_ops.py 15 additions, 0 deletionstests/python/unittest/test_schedule_schedule_ops.py
Loading
Please register or sign in to comment