skip test_mean_half_precision_overflow (#12331)

it only works with SPLIT_REDUCEOP=1
This commit is contained in:
chenyu
2025-09-29 18:15:04 +09:00
committed by GitHub
parent 9915bcf2b4
commit 9d2f2b8e34

View File

@@ -603,6 +603,7 @@ class TestAutoCastType(unittest.TestCase):
t = Tensor([[x]], dtype=dtypes.half, requires_grad=True).expand(N, N).contiguous()
np.testing.assert_allclose(t.mean(axis=1).numpy(), np.array([x] * N, dtype=np.float16), rtol=1e-3)
@unittest.skip("this test only works with SPLIT_REDUCEOP=1")
@unittest.skipUnless(is_dtype_supported(dtypes.half), "need half")
def test_mean_half_precision_overflow(self):
N = 256