[quant] Support for fused ConvBn1d and ConvBnRelu1d modules#38452
[quant] Support for fused ConvBn1d and ConvBnRelu1d modules#38452supriyar wants to merge 6 commits intogh/supriyar/122/basefrom
Conversation
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit 0664c61 (more details on the Dr. CI page):
🕵️ 3 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
| self.bn2 = nn.BatchNorm3d(2).to(dtype=torch.float) | ||
| self.relu3 = nn.ReLU(inplace=True).to(dtype=torch.float) | ||
| self.conv3 = nn.Conv1d(3, 3, 2).to(dtype=torch.float) | ||
| if train: |
There was a problem hiding this comment.
Can you instead check that the code raises an exception for Conv1d + Bn1d and remove the train flag for the model?
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Test Plan: python test/test_quantization.py TestFused Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D21632878](https://our.internmc.facebook.com/intern/diff/D21632878) [ghstack-poisoned]
|
This pull request has been merged in 7d38db0. |
Summary: Test Plan: python test/test_quantization.py TestFused [ghstack-poisoned]
…38452) Summary: Pull Request resolved: pytorch#38452 Test Plan: python test/test_quantization.py TestFused Imported from OSS Differential Revision: D21632878 fbshipit-source-id: 0d73398b95d72a0a23b42ef36f3ede1bfcc35eda
…38452) (pytorch#38749) Summary: Pull Request resolved: pytorch#38749 Test Plan: python test/test_quantization.py TestFused Differential Revision: D21654659 Pulled By: supriyar fbshipit-source-id: 301be24083e794f4e71ff1d6d842e1aaefa640f0
Stack from ghstack:
Summary:
Test Plan:
python test/test_quantization.py TestFused
Reviewers:
Subscribers:
Tasks:
Tags:
Differential Revision: D21632878