Skip to content

Commit 96ca1ee

Browse files
committed
fix discover_coverage
1 parent 2b16530 commit 96ca1ee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/discover_coverage.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -726,9 +726,9 @@ def supports_jvpvjp(self):
726726
return Support.YES
727727
exemptions = {
728728
# we have support (see OpInfo), testing artifact
729-
'torch.nn.functional.dropout2d',
729+
'nn.functional.dropout2d',
730730
# exception: we dont even support double backward for this
731-
'torch.nn.functional.hardswish',
731+
'nn.functional.hardswish',
732732
'bernoulli', # this isn't differentiable
733733
'normal', # not differentiable
734734
}

0 commit comments

Comments
 (0)