search

DeepLink-org / deeplink.framework

BSD 3-Clause "New" or "Revised" License
59 stars 28 forks source link

[DIPU]support torch 2.2 #908

Closed caikun-pjlab closed 3 months ago

caikun-pjlab commented 3 months ago

适配torch 2.2.2 改动点

  1. torch/_dynamo/device_interface.py#L23 校验了Event和Stream的类型，因此DIPU的Event和Stream需要继承_EventBase和_StreamBase
  2. 通信接口支持asyncOp参数
  3. profiler由于移植了不少pytorch代码，改动较多，主要是包含一些基础函数移到c10下需要进行适配 以及 一些函数签名的改变
  4. 注：pytorch2.2的支持仅在nv上进行了测试，没有在寒武纪和华为上进行测试