-
Notifications
You must be signed in to change notification settings - Fork 21.5k
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
[halide-backend] Add GPU support #127506
base: gh/jansel/346/base
Are you sure you want to change the base?
[halide-backend] Add GPU support #127506
Conversation
[ghstack-poisoned]
ghstack-source-id: f916552234a824b197d1601e8b172899b6436d43 Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: 7099c49b82a23722c6402afb786cf753cfe0e37a Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: df5d1c890e0555fd2ac1bcc41208807777980dba Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: 6ea4d6353ee87452cb2976823c7f5152cb7ce3db Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: 51a5eae1047d818867164856540fe539393a4e3b Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: 96bd42bac39a1d70227f987c78f127bc4fa75094 Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: 57baac12e2cffa5556767a8d4d09047be792f4ed Pull Request resolved: #127506
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang [ghstack-poisoned]
ghstack-source-id: f3c533540f1048ef588dc097b5f95114a3e462b7 Pull Request resolved: #127506
ghstack-source-id: 962ebb903bd7bf08aac812e7e808b9e278d83632 Pull Request resolved: #127506
ghstack-source-id: 23ba1db0ebcec8de29ac3138fc31c175a74e58bd Pull Request resolved: #127506
ghstack-source-id: 7a791d8e523fc31585d6cb8b295b8c66e33dc32b Pull Request resolved: pytorch#127506
register_backend_for_device( | ||
"cuda", | ||
CUDACombinedScheduling, | ||
lambda *args, **kwargs: cuda_backends[config.cuda_backend](*args, **kwargs), |
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.
Does simply cuda_backends[config.cuda_backend]
without the lambda wrapper not work?
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.
It fails if you change the config value at runtime.
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Merge failedReason: 2 jobs have failed, first few of them are: trunk / macos-py3-arm64-mps / test (mps, 1, 1, macos-m1-13), trunk / macos-py3-arm64-mps / test (mps, 1, 1, macos-m1-14) Details for Dev Infra teamRaised by workflow job |
ghstack-source-id: 05392d368ee00e3fd793846a1a8ba207adcf3556 Pull Request resolved: pytorch#127506
Stack from ghstack (oldest at bottom):
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang