-
Notifications
You must be signed in to change notification settings - Fork 24.2k
[Inductor][FX]support nn.Linear nn.ConvTransposeNd for efficient_conv_bn_eval #109722
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
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/109722
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 0fee62a with merge base d0c8e82 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@pytorchmergebot 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: 1 jobs have failed, first few of them are: trunk / linux-focal-rocm5.6-py3.8 / test (default, 2, 3, linux.rocm.gpu) Details for Dev Infra teamRaised by workflow job |
@pytorchmergebot 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: 1 jobs have failed, first few of them are: trunk / linux-focal-rocm5.6-py3.8 / test (default, 3, 3, linux.rocm.gpu) Details for Dev Infra teamRaised by workflow job |
@pragupta Can you please take a look at this PR? It's failing on ROCm CI. Please suggest a workaround if the root-causing or fix might take longer. |
Looks like this test (and other similar tests) is disabled via issue #109864, retriggering CI to pick up the skips @pytorchbot rebase |
@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here |
…ose3d for efficient_conv_bn_eval
Successfully rebased |
fd03965
to
0fee62a
Compare
@pytorchmergebot 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 |
Using the
functional_call
API, we can deal with nn.Linear and nn.ConvTransposeNd, just like normal conv.Thanks for @albanD pointing out the API in #109596 .
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @ngimel @yf225 @chenyang78 @kadeng @muchulee8 @aakhundov @jansel