[PASS] Use likely tag & enable LoopPartition by default (#132)
* [PASS] Use likely tag & enable LoopPartition by default * [PASS] Support thread_axis partition * Take IfThenElse branch method * [PASS] Insert branch at the innermost thread scope * [PASS] Select candidates before trying to partition & add test for select * [PASS] Clean code * Fix * Remove print & assert vectorize happens
Showing
- python/tvm/build.py 1 addition, 0 deletionspython/tvm/build.py
- python/tvm/ir_builder.py 14 additions, 0 deletionspython/tvm/ir_builder.py
- src/op/compute_op.cc 7 additions, 6 deletionssrc/op/compute_op.cc
- src/pass/loop_partition.cc 230 additions, 54 deletionssrc/pass/loop_partition.cc
- tests/python/integration/test_ewise.py 2 additions, 2 deletionstests/python/integration/test_ewise.py
- tests/python/unittest/test_codegen_device.py 1 addition, 0 deletionstests/python/unittest/test_codegen_device.py
- tests/python/unittest/test_pass_loop_partition.py 77 additions, 23 deletionstests/python/unittest/test_pass_loop_partition.py
Loading
Please register or sign in to comment