frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

debug handler maintain through decomposition (#141612)

Browse Source 
Add checks in the ao numberic debugger to guard the debug handle consistency between aten op decomposition

Differential Revision: [D66517480](https://our.internmc.facebook.com/intern/diff/D66517480/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/141612
Approved by: https://github.com/jerryzh168
This commit is contained in:
gasoonjia
2024-12-11 18:32:42 -08:00
committed by PyTorch MergeBot
parent 18785c1af9
commit 91261107e0
5 changed files with 158 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/testing/_internal/common_quantization.py
View File

@ -2788,6 +2788,9 @@ class TestHelperModules:
def forward(self, x):
return self.linear2(self.linear1(x))
def example_inputs(self):
return (torch.randn(2, 8),)
class ConvMaxPool2d(torch.nn.Module):
def __init__(self) -> None:
super().__init__()