Github項目推薦 Pytorch TVM 擴展

將TVM集成到PyTorch中。

* TVM：深度學習編譯器

Github項目鏈接：

https://github.com/pytorch/tvm

構建

你需要在這個PR的基礎上構建PyTorch：https://github.com/pytorch/pytorch/pull/18588

然後，你需要單獨構建本倉庫：

測試

使用

如果要禁用JIT掛鉤，請使用 torch_tvm.disable() 。

代碼布局

register.cpp：設置pybind綁定並調用TVM後端的註冊。

compiler.：用TVM編譯PyTorch JIT圖的主要邏輯。

operators.:從JIT IR映射到TVM操作符的位置。

TODO

添加從Python中將不透明op名稱的翻譯註冊到TVM中（如在operator.cpp中完成）的功能。

零拷貝 set_input

紓困機制（調用PyTorch JIT後備）

Threadpool 集成

分配器集成

操作符翻譯

加

乘

卷積

BatchNorm

RELU

AveragePool

MaxPool

線性

張量操作

重塑

查看

備受大家期待的強化學習課程終於上線啦！

掃描下方邀請卡，解鎖更多課時

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！