[API] Prefetch schedule supported (#258)
* prefetch interface added * prefetch python comments modified. prefetch info data structure maintained. * start injecting prefetches. first step (domain touch) implemented. * domain touch tested. * Prefetch ir_mutator and ir_visitor dispatch registered. * modify domain touched from passing a func_ref to passing a tensor * modify domain touched from passing a func_ref to passing a tensor * modify Tensor copy to Tensor ref * temp commit for rebase * debug info removed, typo fixed, ready to rebase * prefetch flatten test add! * roll back builtin functions to side effect functions * lint error fixed! * add cache line size to storage flatten argument * forgot modifications add * change code style to dmlc-like; get rid of can_prove, use manually compute instead * python lint error fixed * modify instrinsic name to pass tests * [TEST] get rid of str(), replace them by accessing attributes * change map to list comprehension * redundant numpy import removed
Showing
- include/tvm/arithmetic.h 13 additions, 0 deletionsinclude/tvm/arithmetic.h
- include/tvm/ir_pass.h 10 additions, 1 deletioninclude/tvm/ir_pass.h
- include/tvm/schedule.h 8 additions, 0 deletionsinclude/tvm/schedule.h
- python/tvm/build.py 2 additions, 1 deletionpython/tvm/build.py
- python/tvm/schedule.py 14 additions, 0 deletionspython/tvm/schedule.py
- src/api/api_arith.cc 8 additions, 1 deletionsrc/api/api_arith.cc
- src/api/api_lang.cc 6 additions, 0 deletionssrc/api/api_lang.cc
- src/api/api_pass.cc 2 additions, 1 deletionsrc/api/api_pass.cc
- src/arithmetic/domain_touched.cc 102 additions, 0 deletionssrc/arithmetic/domain_touched.cc
- src/codegen/llvm/codegen_llvm.cc 1 addition, 1 deletionsrc/codegen/llvm/codegen_llvm.cc
- src/codegen/llvm/intrin_rule_llvm.cc 18 additions, 3 deletionssrc/codegen/llvm/intrin_rule_llvm.cc
- src/op/op_util.cc 3 additions, 3 deletionssrc/op/op_util.cc
- src/pass/inject_prefetch.cc 77 additions, 0 deletionssrc/pass/inject_prefetch.cc
- src/pass/ir_mutator.cc 2 additions, 1 deletionsrc/pass/ir_mutator.cc
- src/pass/ir_visitor.cc 2 additions, 1 deletionsrc/pass/ir_visitor.cc
- src/pass/storage_flatten.cc 65 additions, 3 deletionssrc/pass/storage_flatten.cc
- src/schedule/schedule_lang.cc 18 additions, 0 deletionssrc/schedule/schedule_lang.cc
- tests/python/integration/test_dot.py 1 addition, 1 deletiontests/python/integration/test_dot.py
- tests/python/unittest/test_arith_domain_touched.py 55 additions, 0 deletionstests/python/unittest/test_arith_domain_touched.py
- tests/python/unittest/test_codegen_device.py 1 addition, 1 deletiontests/python/unittest/test_codegen_device.py
Loading
Please register or sign in to comment