Update

ysiraichi · ysiraichi · commit 5ed8aee8548b · 2024-11-11T21:01:40.000-03:00
[ghstack-poisoned]
diff --git a/test/test_meta.py b/test/test_meta.py
@@ -1772,27 +1772,6 @@ def test_triangular_solve_out(self):
         self.assertEqual(out[1].shape, meta_out[1].shape)
         self.assertEqual(out[1].dtype, meta_out[1].dtype)
 
-    def test_meta_consistency_out_dtype_mismatch_pow_Tensor_Scalar(self):
-        S = (5,)
-
-        def run(device):
-            a = torch.rand(S, device=device, dtype=torch.float32)
-            b = 2
-            out = torch.empty(S, device=device, dtype=torch.float64)
-
-            try:
-                torch.pow(a, b, out=out)
-            except Exception as e:
-                return e
-
-        cpu_err = run("cpu")
-        meta_err = run("meta")
-
-        if cpu_err is None and meta_err is not None:
-            raise RuntimeError("cpu didn't fail, but meta did.") from meta_err
-        elif cpu_err is not None and meta_err is None:
-            raise RuntimeError("cpu failed, but meta didn't.") from cpu_err
-
 
 instantiate_device_type_tests(TestMeta, globals())
 
diff --git a/test/test_ops.py b/test/test_ops.py
@@ -207,6 +207,7 @@ def reduction_dtype_filter(op):
     xfail("softmax"),
     xfail("sort"),
     xfail("sparse.sampled_addmm"),
+    xfail("square"),
     xfail("squeeze_copy"),
     xfail("t_copy"),
     xfail("take"),
diff --git a/torch/_refs/__init__.py b/torch/_refs/__init__.py
@@ -1062,7 +1062,6 @@ def _make_elementwise_binary_reference(
     supports_rhs_python_scalar=True,
     supports_two_python_scalars=False,
     should_register_decomposition=True,
-    exact_dtype=False,
 ) -> Callable:
     def inner(prim: Callable):
         nonlocal aten_op, name
@@ -1098,7 +1097,7 @@ def _ref(
             return handle_noncontiguous_outputs([a, b], output)
 
         if has_out:
-            _ref = out_wrapper(exact_dtype=exact_dtype)(_ref)  # type: ignore[assignment]
+            _ref = out_wrapper()(_ref)  # type: ignore[assignment]
 
         _ref.__name__ = name
         if aten_op is infer_aten_op:
@@ -1241,7 +1240,6 @@ def eq(a: TensorLikeType, b: TensorLikeType) -> TensorLikeType:
 
 @_make_elementwise_binary_reference(
     type_promotion_kind=ELEMENTWISE_TYPE_PROMOTION_KIND.BOOL_TO_LONG,
-    exact_dtype=True,
 )
 def pow(
     a: Union[TensorLikeType, NumberType],
