[LANG/SCHEDULE] Reduction factor, predicate in reduction. (#77)
Showing
- include/tvm/expr.h 2 additions, 0 deletionsinclude/tvm/expr.h
- include/tvm/ir.h 1 addition, 1 deletioninclude/tvm/ir.h
- include/tvm/schedule.h 12 additions, 0 deletionsinclude/tvm/schedule.h
- python/tvm/__init__.py 1 addition, 1 deletionpython/tvm/__init__.py
- python/tvm/api.py 15 additions, 6 deletionspython/tvm/api.py
- python/tvm/build.py 60 additions, 13 deletionspython/tvm/build.py
- python/tvm/schedule.py 21 additions, 2 deletionspython/tvm/schedule.py
- src/api/api_ir.cc 1 addition, 1 deletionsrc/api/api_ir.cc
- src/api/api_lang.cc 6 additions, 0 deletionssrc/api/api_lang.cc
- src/codegen/codegen_c.cc 1 addition, 1 deletionsrc/codegen/codegen_c.cc
- src/lang/ir.cc 4 additions, 1 deletionsrc/lang/ir.cc
- src/lang/operation.cc 0 additions, 6 deletionssrc/lang/operation.cc
- src/op/compute_op.cc 6 additions, 30 deletionssrc/op/compute_op.cc
- src/op/op_util.cc 2 additions, 56 deletionssrc/op/op_util.cc
- src/schedule/bound.cc 3 additions, 185 deletionssrc/schedule/bound.cc
- src/schedule/message_passing.cc 343 additions, 0 deletionssrc/schedule/message_passing.cc
- src/schedule/message_passing.h 81 additions, 0 deletionssrc/schedule/message_passing.h
- src/schedule/schedule_dataflow_rewrite.cc 148 additions, 1 deletionsrc/schedule/schedule_dataflow_rewrite.cc
- src/schedule/schedule_lang.cc 50 additions, 17 deletionssrc/schedule/schedule_lang.cc
- tests/python/integration/test_reduce.py 38 additions, 2 deletionstests/python/integration/test_reduce.py
Loading
Please register or sign in to comment