Fix test_llm_config

jackzhxng · facebook-github-bot · commit 1bc2fc13114e · 2025-06-24T17:03:42.000-07:00
Reviewed By: GregoryComer

Differential Revision: D77265031
diff --git a/examples/models/llama/config/test_llm_config.py b/examples/models/llama/config/test_llm_config.py
@@ -94,7 +94,7 @@ def test_valid_llm_config(self):
             backend=BackendConfig(
                 xnnpack=XNNPackConfig(enabled=False),
                 coreml=CoreMLConfig(
-                    enabled=True, ios=17, compute_units=CoreMLComputeUnit.ALL
+                    enabled=True, ios=17, compute_units=CoreMLComputeUnit.cpu_only
                 ),
             ),
         )

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def test_valid_llm_config(self):`
`94`	`94`	`backend=BackendConfig(`
`95`	`95`	`xnnpack=XNNPackConfig(enabled=False),`
`96`	`96`	`coreml=CoreMLConfig(`
`97`		`- enabled=True, ios=17, compute_units=CoreMLComputeUnit.ALL`
	`97`	`+ enabled=True, ios=17, compute_units=CoreMLComputeUnit.cpu_only`
`98`	`98`	`),`
`99`	`99`	`),`
`100`	`100`	`)`