tvm笔记

by

tvm架构

walk through
https://tvm.apache.org/docs/dev/tutorial/codebase_walkthrough.html

runtime system
https://tvm.apache.org/docs/arch/runtime.html#tvm-runtime-system
runtime细节在python/tvm/_ffi/代码里面
PackedFunc机制
https://tvm.apache.org/docs/arch/runtime.html#tvm-runtime-system-packed-func
add op
https://tvm.apache.org/docs/dev/how_to/relay_add_op.html
topi
https://tvm.apache.org/docs/tutorial/intro_topi.html#tutorial-topi

tlcpack的docker添加pudb需要的东西

cp -r pudb pudb-2022.1.3.dist-info /venv/apache-tvm-py3.8/lib/python3.8/site-packages/
cp -r  urwid urwid-2.1.2.dist-info /venv/apache-tvm-py3.8/lib/python3.8/site-packages/
cp -r  urwid_readline urwid_readline-0.14.dist-info /venv/apache-tvm-py3.8/lib/python3.8/site-packages/
cp -r  jedi jedi-0.19.1.dist-info /venv/apache-tvm-py3.8/lib/python3.8/site-packages/
cp -r  parso parso-0.8.4.dist-info /venv/apache-tvm-py3.8/lib/python3.8/site-packages/
cp -r /mast/packageLinux/tvm/tophub/ /root/.tvm/

gemm profiler

src/relay/backend/contrib/cutlass/codegen.cc
python/tvm/contrib/cutlass/build.py
/tvm/tvm/python/tvm/relay/op/contrib/cutlass.py:def partition_for_cutlass(mod, params=None)

Leave a Comment