[Re-landing 86706] [JIT] Frozen Graph Linear-BatchNormNd Folding #91020
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/91020
Note: Links to docs will display an error until the docs builds have been completed. ⏳ No Failures, 2 PendingAs of commit 08fc51d: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@davidberard98 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@davidberard98 looks like all CIs have passed this time! |
davidberard98
left a comment
davidberard98
left a comment
There was a problem hiding this comment.
LGTM! But let me try landing it this time, I'll check internally first
|
@pytorchbot rebase -s |
|
@pytorchbot successfully started a rebase job. Check the current status here |
|
Successfully rebased |
pytorchmergebot
force-pushed
the
minjean/reland_lin_bn_folding
branch
from
3d94c55 to
76acf2d
Compare
|
@davidberard98 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@pytorchbot rebase -s |
|
@pytorchbot successfully started a rebase job. Check the current status here |
|
Successfully rebased |
pytorchmergebot
force-pushed
the
minjean/reland_lin_bn_folding
branch
from
76acf2d to
08fc51d
Compare
|
@davidberard98 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@pytorchbot merge (Initiating merge automatically since Phabricator Diff has merged) |
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 |
5 participants
Re-landing #86706
This PR adds linear-batchnormNd folding for JIT frozen graphs.
Performance benchmark
A preliminary benchmark with a simple model of linear+bn1d tested on first socket, physical cores of skylake machine.
FP32, JIT
without linear-bn folding
with linear-bn folding
