[SCHEDULE][PASS] support storage_align of certain axis (#400)
* [SCHEDULE][PASS] support storage_align of certain axis * fix lint
Showing
- include/tvm/ir.h 7 additions, 0 deletionsinclude/tvm/ir.h
- include/tvm/operation.h 6 additions, 6 deletionsinclude/tvm/operation.h
- include/tvm/schedule.h 17 additions, 0 deletionsinclude/tvm/schedule.h
- python/tvm/schedule.py 20 additions, 0 deletionspython/tvm/schedule.py
- src/api/api_lang.cc 6 additions, 0 deletionssrc/api/api_lang.cc
- src/op/compute_op.cc 21 additions, 5 deletionssrc/op/compute_op.cc
- src/op/extern_op.cc 3 additions, 3 deletionssrc/op/extern_op.cc
- src/op/op_util.cc 1 addition, 1 deletionsrc/op/op_util.cc
- src/op/op_util.h 1 addition, 0 deletionssrc/op/op_util.h
- src/op/placeholder_op.cc 1 addition, 1 deletionsrc/op/placeholder_op.cc
- src/op/scan_op.cc 3 additions, 3 deletionssrc/op/scan_op.cc
- src/pass/storage_flatten.cc 52 additions, 8 deletionssrc/pass/storage_flatten.cc
- src/schedule/schedule_lang.cc 18 additions, 4 deletionssrc/schedule/schedule_lang.cc
- src/schedule/schedule_ops.cc 13 additions, 1 deletionsrc/schedule/schedule_ops.cc
- tests/python/unittest/test_pass_storage_flatten.py 21 additions, 0 deletionstests/python/unittest/test_pass_storage_flatten.py
- tutorials/deployment/cross_compilation_and_rpc.py 6 additions, 8 deletionstutorials/deployment/cross_compilation_and_rpc.py
Loading
Please register or sign in to comment