New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix resolve_name bug #66

Merged

yangbofun merged 2 commits into main from zgc/ditorch_fix_resolve_name_bug

Oct 16, 2024

Collaborator

zhaoguochun1995 commented Oct 16, 2024 •

edited

Loading

torch.randn(3, 4, 5, dtype=torch.float32, device="cuda", requires_grad=True)类算子(如torch.ones_like, torch.full等)实际上是设备上的算子，误被解析为cpu上的算子而被忽略
torch 原生的resolve_name 对有些接口返回None，导致该接口被忽略。返回None的原因是国产框架内使用装饰器对该类算子做了mock，导致返回值异常

zhaoguochun1995 added 2 commits

October 16, 2024 18:05


          fix resolve_name bug

4d5c142


          to_device handle device type args

114a88e

yangbofun approved these changes

View reviewed changes

yangbofun merged commit 1811166 into main

13 checks passed

yangbofun deleted the zgc/ditorch_fix_resolve_name_bug branch

October 16, 2024 10:50

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet