Fix OpInfo definition for mean.

ysiraichi · ysiraichi · commit 9d8ae5fe8329 · 2022-05-12T20:07:23.000+09:00
diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
@@ -17076,18 +17076,17 @@ def error_inputs_mean(op_info, device, **kwargs):
         nan_policy='propagate',
         supports_forward_ad=True,
         supports_fwgrad_bwgrad=True,
+        # FIXME: mean needs 'dim' parameter when using the 'out' overload.
+        # Adding it with 'generate_args_kwargs' does not work, since these also get passed
+        # onto the reference implementations.
+        supports_out=False,
         assert_autodiffed=True,
         assert_jit_shape_analysis=True,
         promotes_int_to_float=True,
         dtypes=floating_and_complex_types_and(torch.float16, torch.bfloat16),
         ref=reference_reduction_numpy(np.mean),
         error_inputs_func=error_inputs_mean,
         skips=(
-            # FIXME: mean needs 'dim' parameter when using the 'out' overload.
-            # Adding it with 'generate_args_kwargs' does not work, since these also get passed
-            # onto the reference implementations.
-            DecorateInfo(unittest.skip("Skipped!"), 'TestCommon', 'test_out'),
-            DecorateInfo(unittest.skip("Skipped!"), 'TestCommon', 'test_out_warning'),
             # FIXME: mean does not support passing keepdim without passing dim
             DecorateInfo(unittest.skip("Skipped!"), 'TestReductions', 'test_dim_default_keepdim'),
             # FIXME: mean reduces all dimensions when dim=[]
