Expectation fixes and added AMD expectations (#38729)

2025-06-13 16:14:58 +02:00
parent e39172ecab
commit 9ff246db00
30 changed files with 311 additions and 146 deletions
--- a/tests/quantization/autoawq/test_awq.py
+++ b/tests/quantization/autoawq/test_awq.py
@@ -62,7 +62,7 @@ class AwqConfigTest(unittest.TestCase):

        # Only cuda and xpu devices can run this function
        support_llm_awq = False
-        device_type, major = get_device_properties()
+        device_type, major, _ = get_device_properties()
        if device_type == "cuda" and major >= 8:
            support_llm_awq = True
        elif device_type == "xpu":