From 1e88edd8c096bde5202dd61411d3c8d7eda56645 Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 25 Jul 2022 17:27:57 -0700 Subject: Removed rowscale (segfaults on ampere). --- tests/test_functional.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'tests/test_functional.py') diff --git a/tests/test_functional.py b/tests/test_functional.py index dcb0255..d80a4f9 100644 --- a/tests/test_functional.py +++ b/tests/test_functional.py @@ -992,6 +992,7 @@ inner = torch.randint(1,4*1024, size=(n,)).tolist() values = list(zip(dim1, dim4, inner)) names = ['dim1_{0}_dim4_{1}_inner_{2}'.format(*vals) for vals in values] @pytest.mark.parametrize("dim1, dim4, inner", values, ids=names) +@pytest.mark.skip("Row scale has some bugs for ampere") def test_igemmlt_row_scale(dim1, dim4, inner): formatB = F.get_special_format_str() err1, err2, err3 = [], [], [] @@ -1064,6 +1065,7 @@ dim4 = [12288, 4096] values = list(zip(dim1, dim4, inner)) names = ['dim1_{0}_dim4_{1}_inner_{2}'.format(*vals) for vals in values] @pytest.mark.parametrize("dim1, dim4, inner", values, ids=names) +@pytest.mark.skip("Row scale has some bugs for ampere") def test_row_scale_bench(dim1, dim4, inner): err1, err2, err3 = [], [], [] relerr1, relerr2 = [], [] -- cgit v1.2.3