From 8cf057ba09cbe6f1dfb4d469b2d49f230ec5cc0f Mon Sep 17 00:00:00 2001 From: shanjiaz Date: Thu, 22 May 2025 10:37:07 -0400 Subject: [PATCH 1/4] add observer type for test_min_max Signed-off-by: shanjiaz --- tests/llmcompressor/observers/test_min_max.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llmcompressor/observers/test_min_max.py b/tests/llmcompressor/observers/test_min_max.py index b592579f6..978e3fac6 100644 --- a/tests/llmcompressor/observers/test_min_max.py +++ b/tests/llmcompressor/observers/test_min_max.py @@ -35,7 +35,7 @@ def make_dummy_g_idx(columns: int, group_size: int) -> torch.Tensor: def test_min_max_observer(symmetric, expected_scale, expected_zero_point): tensor = torch.tensor([1, 1, 1, 1, 1]) num_bits = 8 - weights = QuantizationArgs(num_bits=num_bits, symmetric=symmetric) + weights = QuantizationArgs(num_bits=num_bits, symmetric=symmetric, observer="minmax") observer = weights.observer observer = Observer.load_from_registry(observer, quantization_args=weights) From 991faa01019716bf0c453fd4253c35b9f88deea6 Mon Sep 17 00:00:00 2001 From: shanjiaz Date: Thu, 22 May 2025 10:42:56 -0400 Subject: [PATCH 2/4] fix style Signed-off-by: shanjiaz --- tests/llmcompressor/observers/test_min_max.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/llmcompressor/observers/test_min_max.py b/tests/llmcompressor/observers/test_min_max.py index 978e3fac6..95ed1f1e0 100644 --- a/tests/llmcompressor/observers/test_min_max.py +++ b/tests/llmcompressor/observers/test_min_max.py @@ -35,7 +35,9 @@ def make_dummy_g_idx(columns: int, group_size: int) -> torch.Tensor: def test_min_max_observer(symmetric, expected_scale, expected_zero_point): tensor = torch.tensor([1, 1, 1, 1, 1]) num_bits = 8 - weights = QuantizationArgs(num_bits=num_bits, symmetric=symmetric, observer="minmax") + weights = QuantizationArgs(num_bits=num_bits, + symmetric=symmetric, + observer="minmax") observer = weights.observer observer = Observer.load_from_registry(observer, quantization_args=weights) From 7be4c0087e244bb86342cd74874410e9ce1bd443 Mon Sep 17 00:00:00 2001 From: shanjiaz Date: Thu, 22 May 2025 11:50:04 -0400 Subject: [PATCH 3/4] commit to bump ci/cd Signed-off-by: shanjiaz --- tests/llmcompressor/observers/test_min_max.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/llmcompressor/observers/test_min_max.py b/tests/llmcompressor/observers/test_min_max.py index 95ed1f1e0..97d3391b2 100644 --- a/tests/llmcompressor/observers/test_min_max.py +++ b/tests/llmcompressor/observers/test_min_max.py @@ -35,6 +35,7 @@ def make_dummy_g_idx(columns: int, group_size: int) -> torch.Tensor: def test_min_max_observer(symmetric, expected_scale, expected_zero_point): tensor = torch.tensor([1, 1, 1, 1, 1]) num_bits = 8 + weights = QuantizationArgs(num_bits=num_bits, symmetric=symmetric, observer="minmax") From 0c6bc6fe0ac8ed3c9249fcaf51eb7ddf14e7fbda Mon Sep 17 00:00:00 2001 From: shanjiaz Date: Thu, 22 May 2025 11:53:24 -0400 Subject: [PATCH 4/4] fix style Signed-off-by: shanjiaz --- tests/llmcompressor/observers/test_min_max.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llmcompressor/observers/test_min_max.py b/tests/llmcompressor/observers/test_min_max.py index 97d3391b2..2127f37b5 100644 --- a/tests/llmcompressor/observers/test_min_max.py +++ b/tests/llmcompressor/observers/test_min_max.py @@ -35,7 +35,7 @@ def make_dummy_g_idx(columns: int, group_size: int) -> torch.Tensor: def test_min_max_observer(symmetric, expected_scale, expected_zero_point): tensor = torch.tensor([1, 1, 1, 1, 1]) num_bits = 8 - + weights = QuantizationArgs(num_bits=num_bits, symmetric=symmetric, observer="minmax")