Merge nucleic/sleek-ember-seal-uady into dev

2026-07-30 21:47:07 -07:00
parent a374fa40b6
commit 82545b833b
3 changed files with 16 additions and 5 deletions
@@ -37,12 +37,15 @@ class FakeOptimize:
 class CoreMLQuantizationConfigTests(unittest.TestCase):
    def test_matches_accepted_qdq_policy(self):
        activation, weights = quantize_coreml._optimization_configs(FakeOptimize)
-        self.assertEqual("linear", activation.global_config.values["mode"])
-        self.assertIs(np.uint8, activation.global_config.values["dtype"])
+        self.assertIsNone(activation.global_config)
+        activation_linear = activation.op_type_configs["linear"].values
+        self.assertEqual("linear", activation_linear["mode"])
+        self.assertIs(np.uint8, activation_linear["dtype"])
        self.assertEqual(
            "per_tensor",
-            activation.global_config.values["granularity"],
+            activation_linear["granularity"],
        )
+        self.assertEqual({"linear"}, set(activation.op_type_configs))

        linear = weights.op_type_configs["linear"].values
        self.assertEqual("linear_symmetric", linear["mode"])