issues
search
DeepLink-org
/
deeplink.framework
BSD 3-Clause "New" or "Revised" License
55
stars
28
forks
source link
[DIPU]support torch 2.2
#908
Closed
caikun-pjlab
closed
1 month ago
caikun-pjlab
commented
1 month ago
适配torch 2.2.2 改动点
torch/_dynamo/device_interface.py#L23
校验了Event和Stream的类型,因此DIPU的Event和Stream需要继承_EventBase和_StreamBase
通信接口支持asyncOp参数
profiler由于移植了不少pytorch代码,改动较多,主要是包含一些基础函数移到c10下需要进行适配 以及 一些函数签名的改变
注:pytorch2.2的支持仅在nv上进行了测试,没有在寒武纪和华为上进行测试
适配torch 2.2.2 改动点