Skip to content

Commit

Permalink
.
Browse files Browse the repository at this point in the history
  • Loading branch information
[email protected] committed Nov 4, 2024
1 parent 8a195cd commit 7433739
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/test_general_reduction_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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)
Expand Down

0 comments on commit 7433739

Please sign in to comment.