-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Tests Generelization for multiple accelerator devices #135242
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.
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
Tests Generelization for multiple accelerator devices #135242
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/135242
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 4aec93c with merge base 228963a ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hello, this might optimized to pythonic expression, like to_device = self.rank if TEST_CUDA else device I think. :D
And I saw there's some repeating in this logic, extract a method might be an option to reduce duplication, and convenient for other device if they want to reuse test-cases as well. Thanks
|
/easycla |
|
@rahulsingh-intel : can you please address this change pointed out by @awgu to the PR after rebase to main. #133209 (comment). Thanks |
4712e89 to
df5bbc0
Compare
ec978d3 to
4aec93c
Compare
Motivation: Generalize unit tests so that can be executed for cuda and non cuda devices. Depedency : #133209 Merged now. There was a #135242 for these changes and closed due to in correct commits. I have incoroprated the changes as suggested in comments. @kwen2501 @zeshengzong Please review the changes. Pull Request resolved: #139184 Approved by: https://github.com/kwen2501 Co-authored-by: Yu, Guangye <[email protected]>
Motivation: Generalize unit tests so that can be executed for cuda and non cuda devices. Depedency : #133209 Merged now. There was a #135242 for these changes and closed due to in correct commits. I have incoroprated the changes as suggested in comments. @kwen2501 @zeshengzong Please review the changes. Pull Request resolved: #139184 Approved by: https://github.com/kwen2501 Co-authored-by: Yu, Guangye <[email protected]>
Motivation: Generalize unit tests so that can be executed for cuda and non cuda devices.
Brief: Changes have depedency on #133209.
cc @H-Huang @awgu @kwen2501 @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @c-p-i-o @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10 @gujinghui @PenghuiCheng @jianyuh @min-jean-cho @yanbing-j @Guobing-Chen @Xia-Weiwen @snadampal @EikanWang @voznesenskym @penguinwu @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang @aakhundov @rec @xmfan @XilunWu