Skip to content

Commit 3559da7

Browse files
committed
Added a failing test case
1 parent 0d4af77 commit 3559da7

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

tests/py/dynamo/conversion/test_group_norm_aten.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,27 @@ def forward(self, x):
112112
inputs,
113113
)
114114

115+
def test_groupnorm_sd(self):
116+
class GroupNorm(torch.nn.Module):
117+
def forward(self, x):
118+
return torch.ops.aten.native_group_norm.default(
119+
x,
120+
torch.randn((320,)).half(),
121+
torch.randn((320,)).half(),
122+
2,
123+
320,
124+
4096,
125+
32,
126+
1e-05,
127+
)[0]
128+
129+
inputs = [torch.randn(2, 320, 64, 64).half()]
130+
with torch.no_grad():
131+
self.run_test(
132+
GroupNorm(),
133+
inputs,
134+
)
135+
115136
@parameterized.expand(
116137
[
117138
(5, 4, 4, 2, (2, 4, 2), (3, 4, 2), (5, 4, 4)),

0 commit comments

Comments
 (0)