projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 22db299) | patch
[TIR][REFACTOR] RewriteForTensorCore -> te/schedule (#5379)
authorTianqi Chen <tqchen@users.noreply.github.com>
Mon, 20 Apr 2020 02:57:25 +0000 (19:57 -0700)
committerGitHub <noreply@github.com>
Mon, 20 Apr 2020 02:57:25 +0000 (19:57 -0700)
commit4c0a53dc5bef49797ccbb5f05b0c61ca832d7c85
tree840310de1719a5a1033d13c2d36329bcc94b9d21tree | snapshot
parent22db299b33f05570db2a5a406bdb37b57198a822commit | diff
[TIR][REFACTOR] RewriteForTensorCore -> te/schedule (#5379)

* [TIR][REFACTIR] RewriteForTensorCore -> te/schedule

RewriteForTensor depends on the schedule information, which makes it differ
from a typical pass(which should get all the information from the input TIR).

As a result, we refactor it as a SchedulePostProc step for now.
We should revisit it later as we introduce more support for tensor core patterns in the TIR.

* Fix VTA to fit the new IR Pattern
include/tvm/te/schedule_pass.h diff | blob | history
include/tvm/tir/ir_pass.h diff | blob | history
python/tvm/driver/build_module.py diff | blob | history
python/tvm/te/hybrid/__init__.py diff | blob | history
src/te/schedule/schedule_postproc_rewrite_for_tensor_core.cc [moved from src/tir/pass/tensor_core.cc with 97% similarity] diff | blob | history
src/tir/pass/ffi_api.cc diff | blob | history
vta/python/vta/ir_pass.py diff | blob | history
Domain: Machine Learning / ML Framework;