写点什么

pytroch

0 人感兴趣 · 2 次引用

  • 最新
  • 推荐
https://static001.geekbang.org/infoq/5c/5c7a766159c7779e5466ebe6048dea7f.png?x-oss-process=image%2Fresize%2Cw_416%2Ch_234

Ascend 的 aclgraph(一)aclgraph 是什么?torchair 又是怎么成图的?

用户头像
zjun
17 小时前

PyTorch框架默认采用Eager模式,单算子下发后立即执行,每个算子都需要从Host侧Python API->Host侧C++层算子下发->Device侧算子kernel执行,在Device侧每次kernel执行之前都需要等待Host侧的下发逻辑完成。因此当单个算子计算量过小或者Host性能不佳时,很容

pytroch_pytroch技术文章_InfoQ写作社区