File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -597,6 +597,8 @@ class Support(enum.Enum):
597
597
'nn.functional.dropout' , # not actually problem, randomness testing artifact
598
598
'nn.functional.dropout2d' , # not actually problem, randomness testing artifact
599
599
'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
600
602
}
601
603
602
604
VMAP_EXEMPTIONS = {
@@ -862,6 +864,9 @@ def summary(self):
862
864
# pprint.pprint(result)
863
865
# result = opset.query(Operator.supports_jvpvjp, (Support.NO, Support.UNKNOWN))
864
866
# pprint.pprint(result)
867
+ print ("supports_vjp" )
868
+ result = opset .query (Operator .supports_vjp , (Support .NO , Support .UNKNOWN ))
869
+ pprint .pprint (result )
865
870
print ("supports_jvp" )
866
871
result = opset .query (Operator .supports_jvp , (Support .NO , Support .UNKNOWN ))
867
872
pprint .pprint (result )
You can’t perform that action at this time.
0 commit comments