[aotinductor] Enable test_non_default_cuda_device on CI #110509
Conversation
Summary: test_non_default_cuda_device needs to run on a multi-gpu CI instance [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/110509
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit a4da0e7 with merge base 46a5558 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
| @requires_multigpu() | ||
| def test_non_default_cuda_device(self): | ||
| if self.device == "cpu": | ||
| raise unittest.SkipTest("requires CUDA") |
There was a problem hiding this comment.
Wondering if @requires_cuda() decorator would work or not?
There was a problem hiding this comment.
That will be a problem when running on a gpu instance, where both cuda and cpu version of this test will be run.
Summary: test_non_default_cuda_device needs to run on a multi-gpu CI instance cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
|
@desertfire has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@pytorchbot merge -f 'Landed internally' (Initiating merge automatically since Phabricator Diff has merged, using force because this PR might not pass merge_rules.json but landed internally) |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
6 participants
Stack from ghstack (oldest at bottom):
Summary: test_non_default_cuda_device needs to run on a multi-gpu CI instance
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @aakhundov @ColinPeppler
Differential Revision: D49937115