专栏文章 2020.5

magicyang · 2020年05月22日 · 393 次阅读

最近在做什么?

华为 Atlas 算法适配。
一路趟雷下来。
用各种奇葩的办法去规避一个奇葩的硬件。
3 个月下来彻底打破了我原来对华为的认识。。。(以前华为是稳定不出错著称的。。。)
哎,要不是领导死压,根本不可能做的下去。

做了什么?

因为我们是 Pytorch 的框架,华为只支持 tensorflow 和 caffe.
基于开源改了一个 onnx->tensorflow 的工具,为了规避华为的各种算子限制,修改了不少算子。
开始看华为的框架跟天书一样,舔着脸去论坛问各种小白问题,现在总算是基本的坑都趟了一遍。

个人所得

以前就是找合适的算子训练,但是有什么坑,到底怎么运算的,基础其实有点差。
半写了一套模型转化框架,基本的算子操作方式,tensorflow,pytorch 的 OPS 是有些熟悉了。
填坑,尤其是填巨坑是一件很痛苦的事情,但是个人确实成长快。

不满意的地方

有些人和事真是一言难尽。。。

方向

未来希望继续做算法和硬件适配吧,有个东西叫 TVM,希望未来能把这块做好吧。

暫無回覆。
需要 登录 後方可回應,如果你還沒有帳號按這裡 注册