[Ansor][AutoTVM v2.0] Phase 1: Add follow_split and follow_fused_split steps (#6142)
* Add cache_read/cache_write step
* Update
* Add follow split and follow fused split
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
Conflicts:
src/auto_scheduler/compute_dag.cc
src/auto_scheduler/transform_step.cc
src/auto_scheduler/transform_step.h
tests/python/unittest/test_auto_scheduler_loop_state.py
* add loop_state.py
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
* Update
* Update state->current_compute_dag to Optional
* Add some doc strings for Follow_Split and Follow_fused_split
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Check code using c-lint
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add more doc strings and change the order for follow split.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add record test for follow_split and follow_fused_split
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add record test for follow_split
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add record test for follow_fused_split.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add test record for follow_fused_split
1. delete a comment
2. add "fuse" between follow_split and follow_fused_split
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add doc strings for some functions and variables
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Fix the code format in src/auto_scheduler/transform_step.h
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
* Update doc
* Update
* Update
* Fix follow_split and follow_fused_split record test.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Doc update
* Update some doc strings
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Fix code style and some function definitions.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add comments on parameters.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Add more doc strings and fix some.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
* Update.
Signed-off-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>
Co-authored-by: chengfan.jcf <chengfan.jcf@alibaba-inc.com>
Co-authored-by: jingbang.yjb <jingbang.yjb@alibaba-inc.com>