Skip to content

Commit 5170948

Browse files
committed
Update discover_coverage
1 parent 23b59b0 commit 5170948

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/discover_coverage.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -597,6 +597,8 @@ class Support(enum.Enum):
597597
'nn.functional.dropout', # not actually problem, randomness testing artifact
598598
'nn.functional.dropout2d', # not actually problem, randomness testing artifact
599599
'nn.functional.rrelu', # not actually problem, randomness testing artifact
600+
'bernoulli', # not actually problem, randomness testing artifact
601+
'normal', # not actually problem, randomness testing artifact
600602
}
601603

602604
VMAP_EXEMPTIONS = {
@@ -862,6 +864,9 @@ def summary(self):
862864
# pprint.pprint(result)
863865
# result = opset.query(Operator.supports_jvpvjp, (Support.NO, Support.UNKNOWN))
864866
# pprint.pprint(result)
867+
print("supports_vjp")
868+
result = opset.query(Operator.supports_vjp, (Support.NO, Support.UNKNOWN))
869+
pprint.pprint(result)
865870
print("supports_jvp")
866871
result = opset.query(Operator.supports_jvp, (Support.NO, Support.UNKNOWN))
867872
pprint.pprint(result)

0 commit comments

Comments
 (0)