From 74337398e521c899fa4c91adf5ecece976aa59b7 Mon Sep 17 00:00:00 2001 From: "hanhaowen@sensetime.com" Date: Mon, 4 Nov 2024 09:16:14 +0000 Subject: [PATCH] . --- tests/test_general_reduction_ops.py | 2 ++ 1 file changed, 2 insertions(+) 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)