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

[Tensor] fix optimizer for CPU parallel #1069

Merged

Wesley-Jzy merged 1 commit into hpcaitech:main from Wesley-Jzy:tensor/fix_cpu_parallel

Jun 6, 2022

Contributor

Wesley-Jzy commented Jun 6, 2022

CPU DP will meet segment fault because CPU is not protected by stream. Since torch has no CPU stream, I directly return grad for CPU grad.


          fix optimizer for CPU parallel

87f344d

Wesley-Jzy added the Run Build and Test label

Wesley-Jzy requested review from feifeibear and ver217

June 6, 2022 09:30

ver217 approved these changes

View reviewed changes

Wesley-Jzy removed the Run Build and Test label

Wesley-Jzy merged commit 4fc748f into hpcaitech:main

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