From 093602cb27f92aa371104ebbf8cc2f3f487e9fce Mon Sep 17 00:00:00 2001 From: Xin He Date: Wed, 22 Jan 2025 16:49:02 +0800 Subject: [PATCH] disable conv1d for rtn on HPU --- test/3x/torch/quantization/weight_only/test_rtn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/3x/torch/quantization/weight_only/test_rtn.py b/test/3x/torch/quantization/weight_only/test_rtn.py index 4f2b9c44752..352e0246001 100644 --- a/test/3x/torch/quantization/weight_only/test_rtn.py +++ b/test/3x/torch/quantization/weight_only/test_rtn.py @@ -309,6 +309,8 @@ def test_rtn_with_quantize_API(self): ), "The results of calling `convert` + `prepare` and calling `quantize` should be equal." # TODO: (4, True, 32, 0), group_dim=0, format not supported + # TODO [SW-216127]: it's not in high priority, so we can implement it later. + @pytest.mark.skipif(is_hpex_available(), reason="These tests are not supported on HPU for now.") @pytest.mark.parametrize( "bits, use_sym, group_size, group_dim", [