[SCHEDULE] Enhance cache_write to enable layout change. (#432)
* [SCHEDULE] Enahance cache_write to enable layout change. * more tests
Showing
- include/tvm/schedule.h 9 additions, 2 deletionsinclude/tvm/schedule.h
- python/tvm/schedule.py 8 additions, 0 deletionspython/tvm/schedule.py
- src/op/compute_op.cc 5 additions, 4 deletionssrc/op/compute_op.cc
- src/op/cross_thread_reduction.cc 3 additions, 3 deletionssrc/op/cross_thread_reduction.cc
- src/op/op_util.cc 0 additions, 85 deletionssrc/op/op_util.cc
- src/op/op_util.h 1 addition, 16 deletionssrc/op/op_util.h
- src/op/scan_op.cc 1 addition, 1 deletionsrc/op/scan_op.cc
- src/schedule/message_passing.cc 138 additions, 2 deletionssrc/schedule/message_passing.cc
- src/schedule/message_passing.h 32 additions, 0 deletionssrc/schedule/message_passing.h
- src/schedule/schedule_dataflow_rewrite.cc 120 additions, 34 deletionssrc/schedule/schedule_dataflow_rewrite.cc
- tests/python/unittest/test_schedule_schedule_ops.py 51 additions, 1 deletiontests/python/unittest/test_schedule_schedule_ops.py
Loading
Please register or sign in to comment