Revert "Update torch.masked.mean to upcast dtype for bool tensors (pytorch#139999)"

This reverts commit 0742b23.

Reverted pytorch#139999 on behalf of https://github.com/huydhn due to Sorry for reverting your change but I think it has a landrace and fails a test in trunk ([comment](pytorch#139999 (comment)))

Author: pytorchmergebot

torch/masked/_ops.py

@@ -1384,16 +1384,8 @@ def mean(
 {reduction_args}
 
 {reduction_example}"""
-    dtype_source = "Optional"
     if dtype is None:
         dtype = input.dtype
-        dtype_source = "Input"
-
-    if not (dtype.is_floating_point or dtype.is_complex):
-        raise ValueError(
-            f"mean(): Could not infer output dtype. {dtype_source} dtype must be either "
-            f"a floating point or complex dtype. Got: {dtype}"
-        )
     if input.layout == torch.strided:
         if mask is None:
             # TODO: compute count analytically

torch/testing/_internal/opinfo/definitions/_masked.py

@@ -769,8 +769,26 @@ def sample_inputs_masked_normalize(op_info, device, dtype, requires_grad, **kwar
         supports_forward_ad=True,
         supports_fwgrad_bwgrad=True,
         promotes_int_to_float=True,
-        dtypes=floating_and_complex_types_and(torch.float16, torch.bfloat16),
+        dtypes=all_types_and_complex_and(torch.float16, torch.bfloat16, torch.bool),
         skips=(
+            DecorateInfo(
+                unittest.expectedFailure,
+                "TestReductions",
+                "test_ref_duplicate_values",
+                dtypes=(torch.bool,),
+            ),
+            DecorateInfo(
+                unittest.expectedFailure,
+                "TestReductions",
+                "test_reference_masked",
+                dtypes=(torch.bool,),
+            ),
+            DecorateInfo(
+                unittest.expectedFailure,
+                "TestReductions",
+                "test_ref_small_input",
+                dtypes=(torch.bool,),
+            ),
             DecorateInfo(
                 unittest.expectedFailure,
                 "TestNormalizeOperators",