File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -623,6 +623,12 @@ def vjp_of_vjp(*args_and_cotangents):
623
623
624
624
@ops (functorch_lagging_op_db + additional_op_db , allowed_dtypes = (torch .float ,))
625
625
@toleranceOverride ({torch .float32 : tol (atol = 1e-04 , rtol = 1e-04 )})
626
+ @opsToleranceOverride ('TestOperators' , 'test_vmapvjp' , (
627
+ tol1 ('linalg.svd' ,
628
+ {torch .float32 : tol (atol = 1.5e-04 , rtol = 1e-04 )}, device_type = "cuda" ),
629
+ tol1 ('svd' ,
630
+ {torch .float32 : tol (atol = 1.5e-04 , rtol = 1e-04 )}, device_type = "cuda" ),
631
+ ))
626
632
@skipOps ('TestOperators' , 'test_vmapvjp' , vmapvjp_fail )
627
633
def test_vmapvjp (self , device , dtype , op ):
628
634
if not op .supports_autograd :
@@ -1031,7 +1037,6 @@ def test():
1031
1037
# fallback path doesn't work
1032
1038
# All of the following are bugs and need to be fixed
1033
1039
xfail ('__getitem__' , '' ),
1034
- xfail ('clamp' , '' ),
1035
1040
xfail ('index_put' , '' ),
1036
1041
xfail ('matrix_exp' ),
1037
1042
xfail ('view_as_complex' ),
You can’t perform that action at this time.
0 commit comments