diff --git a/tests/test_general_reduction_ops.py b/tests/test_general_reduction_ops.py index 0662589d..8dcaa411 100644 --- a/tests/test_general_reduction_ops.py +++ b/tests/test_general_reduction_ops.py @@ -122,6 +122,7 @@ def test_accuracy_max_without_dim(shape, dtype): gems_assert_equal(res_out, ref_out) + @pytest.mark.max @pytest.mark.parametrize("shape", REDUCTION_SHAPES) @pytest.mark.parametrize("dtype", FLOAT_DTYPES) @@ -135,6 +136,7 @@ def test_accuracy_max_without_dim_uncontiguous(shape, dtype): gems_assert_equal(res_out, ref_out) + # TODO: failed at (200, 40999, 3), while successed at this shape in mean_dim @pytest.mark.max @pytest.mark.parametrize("shape", REDUCTION_SMALL_SHAPES)