Skip to content

Commit 1f80805

Browse files
committed
MAINT: mention the pytorch issue in mean(..., dtype=float16) test
1 parent bfef3a4 commit 1f80805

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_np/tests/test_reductions.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,7 @@ def test_mean_values(self):
259259
res = np.mean(mat, axis=axis) * mat.size
260260
assert_allclose(res, tgt)
261261

262-
@pytest.mark.xfail(reason="XXX: mean(..., dtype=..) smarter dtype selection")
262+
@pytest.mark.xfail(reason="see pytorch/gh-91597")
263263
def test_mean_float16(self):
264264
# This fail if the sum inside mean is done in float16 instead
265265
# of float32.

0 commit comments

Comments
 (0)