Masahiro Masuda 3823b39b8a [TIR] Utility function to decide loop mapping for auto tensorization (#11050) ...

* [TIR] Add TensorizeInfo and GetTensorizeLoopMapping

* expose PreOrderVisit to python

* add test case

* add conv2d nchwc test

* add mma test

* add arm nhwc conv2d test

* Revert "add arm nhwc conv2d test"

This reverts commit eb147f33bb02d62a0eacc9cdfe777ac047ee1bc9.

* refine

* add doc

* update

* fixd condition

* black

* pylint

* Update python/tvm/tir/schedule/analysis.py

Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>

* run black

* bring back logic in original code to support loop permutation

* add comment

* simplify

* minor fix to test

Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

2022-04-19 20:01:24 -07:00

..

tvm

[TIR] Utility function to decide loop mapping for auto tensorization (#11050)

2022-04-19 20:01:24 -07:00

.gitignore

Generate requirements.txt from Python spec (#7289)

2021-02-04 06:51:12 -08:00

gen_requirements.py

[CI][Caffe Frontend] Change the caffe deps into SSD distribution (#9060)

2022-01-03 15:19:33 -08:00

setup.py

[microTVM] Add microTVM Template Projects to tlcpack pip Package (#9309)

2021-11-01 17:16:23 +00:00