Fix CI (#844)

zou3519 · web-flow · commit 23b59b0716d0 · 2022-06-01T09:54:16.000-04:00
diff --git a/functorch/_src/eager_transforms.py b/functorch/_src/eager_transforms.py
@@ -1327,7 +1327,5 @@ def _register_python_decomposition_vmap(decomp):
 _register_jit_decomposition(torch.ops.aten._log_softmax_backward_data.default)
 _register_jit_decomposition(torch.ops.aten._softmax_backward_data.default)
 _register_jit_decomposition(torch.ops.aten.log_sigmoid_forward.default)
-_register_jit_decomposition(torch.ops.aten.binary_cross_entropy_backward.default)
-_register_jit_decomposition(torch.ops.aten.binary_cross_entropy.default)
 _register_jit_decomposition(torch.ops.aten.native_layer_norm_backward.default)
 _register_python_decomposition_vmap(torch.ops.aten.addr.default)
diff --git a/functorch/csrc/DynamicLayer.cpp b/functorch/csrc/DynamicLayer.cpp
@@ -479,8 +479,6 @@ TORCH_LIBRARY_IMPL(aten, FT_DYNAMIC_LAYER_FRONT_MODE_KEY, m) {
   JVP_DECOMP(_softmax_backward_data);
   OP_DECOMPOSE(log_sigmoid);
   JVP_DECOMP(log_sigmoid_forward);
-  JVP_DECOMP(binary_cross_entropy);
-  JVP_DECOMP(binary_cross_entropy_backward);
   JVP_DECOMP(native_layer_norm_backward);
 }
 
diff --git a/test/test_ops.py b/test/test_ops.py
@@ -1248,7 +1248,6 @@ def reference(primals, cotangents, primals_tangents, cotangents_tangents):
                 'softmax',
                 'log_softmax',
                 'nn.functional.cross_entropy',
-                'nn.functional.binary_cross_entropy',
                 'nn.functional.layer_norm'
             }
             if op.name in FUNCTORCH_HAS_FORMULA_BUT_NOT_PYTORCH:

Original file line number	Diff line number	Diff line change
`@@ -479,8 +479,6 @@ TORCH_LIBRARY_IMPL(aten, FT_DYNAMIC_LAYER_FRONT_MODE_KEY, m) {`
`479`	`479`	`JVP_DECOMP(_softmax_backward_data);`
`480`	`480`	`OP_DECOMPOSE(log_sigmoid);`
`481`	`481`	`JVP_DECOMP(log_sigmoid_forward);`
`482`		`- JVP_DECOMP(binary_cross_entropy);`
`483`		`- JVP_DECOMP(binary_cross_entropy_backward);`
`484`	`482`	`JVP_DECOMP(native_layer_norm_backward);`
`485`	`483`	`}`
`486`	`484`
Original file line number	Diff line number	Diff line change
`@@ -1248,7 +1248,6 @@ def reference(primals, cotangents, primals_tangents, cotangents_tangents):`
`1248`	`1248`	`'softmax',`
`1249`	`1249`	`'log_softmax',`
`1250`	`1250`	`'nn.functional.cross_entropy',`
`1251`		`- 'nn.functional.binary_cross_entropy',`
`1252`	`1251`	`'nn.functional.layer_norm'`
`1253`	`1252`	`}`
`1254`	`1253`	`if op.name in FUNCTORCH_HAS_FORMULA_BUT_NOT_PYTORCH:`