[Hybrid Script] Supporting scheduling hybrid script (#2416)
* on the way to enable hybrid schedule * I think I am done with imperfect loop split? * copyright watermark * loop annotation * fix lint * fix lint 1 * shit! * loop reorder supported * support bind to add some tests * fused tested * imperfect loop testcase * fix lint * add bind testcase * fix doc * fix online edit typo * resolve @mercymercy review * fix indent * i should convince myself it is not flaky test first * fix test hybrid * how many flaky test are you expecting; i ball ball u to let me pass * rebase halide...
Showing
- docs/langref/hybrid_script.rst 9 additions, 3 deletionsdocs/langref/hybrid_script.rst
- include/tvm/operation.h 3 additions, 0 deletionsinclude/tvm/operation.h
- python/tvm/tensor.py 5 additions, 2 deletionspython/tvm/tensor.py
- src/op/compute_op.cc 1 addition, 0 deletionssrc/op/compute_op.cc
- src/op/hybrid_op.cc 321 additions, 14 deletionssrc/op/hybrid_op.cc
- src/op/hybrid_op.h 80 additions, 0 deletionssrc/op/hybrid_op.h
- src/op/op_util.cc 30 additions, 32 deletionssrc/op/op_util.cc
- src/op/op_util.h 12 additions, 8 deletionssrc/op/op_util.h
- tests/python/unittest/test_hybrid_script.py 107 additions, 9 deletionstests/python/unittest/test_hybrid_script.py
Loading
Please register or sign in to comment