[FUSION] add 'void AutoFuseEwise(Schedule sch)' (#36)
* [FUSION] add Fusion(Schedule) * [FUSION] rename to AutoFuseEwise, detect whether the stage has been scheduled * [FUSION] change to visitor pattern * [FUSION] rename filename * [FUSION] fine-tune the interface * [FUSION] typo * move elem_wise to schedule * rename test function
Showing
- include/tvm/ir_pass.h 0 additions, 1 deletioninclude/tvm/ir_pass.h
- include/tvm/schedule.h 11 additions, 0 deletionsinclude/tvm/schedule.h
- include/tvm/schedule_pass.h 7 additions, 0 deletionsinclude/tvm/schedule_pass.h
- python/tvm/schedule.py 1 addition, 1 deletionpython/tvm/schedule.py
- src/api/api_schedule.cc 5 additions, 0 deletionssrc/api/api_schedule.cc
- src/schedule/auto_inline_elem_wise.cc 76 additions, 0 deletionssrc/schedule/auto_inline_elem_wise.cc
- tests/python/unittest/test_schedule_schedule_ops.py 16 additions, 0 deletionstests/python/unittest/test_schedule_schedule_ops.py
Loading
Please register or sign in to comment