skip one op for coreml partitioner (#2826)

cccclai · facebook-github-bot · commit 549001cd16f6 · 2024-04-04T13:44:40.000-07:00
Summary:

There are some issues with the current two ops in coreml side. While waiting for the fix, skip them for now 
```
"aten.index.tensor", "aten.index_put.default"
```
test with
```
python3 -m examples.models.llama2.export_llama --coreml --use_kv_cache
```

Reviewed By: mergennachin

Differential Revision: D55680297
diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py
@@ -593,7 +593,10 @@ def _export_llama(modelname, args) -> str:  # noqa: C901
         partitioners.append(
             # pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `apple`
             CoreMLPartitioner(
-                skip_ops_for_coreml_delegation=None, compile_specs=compile_specs
+                skip_ops_for_coreml_delegation=[
+                    "aten.index_put.default",
+                ],
+                compile_specs=compile_specs,
             )
         )
         modelname = f"coreml_{modelname}"

Original file line number	Diff line number	Diff line change
`@@ -593,7 +593,10 @@ def _export_llama(modelname, args) -> str: # noqa: C901`
`593`	`593`	`partitioners.append(`
`594`	`594`	# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `apple`
`595`	`595`	`CoreMLPartitioner(`
`596`		`- skip_ops_for_coreml_delegation=None, compile_specs=compile_specs`
	`596`	`+ skip_ops_for_coreml_delegation=[`
	`597`	`+ "aten.index_put.default",`
	`598`	`+ ],`
	`599`	`+ compile_specs=compile_specs,`
`597`	`600`	`)`
`598`	`601`	`)`
`599`	`602`	`modelname = f"coreml_{modelname}"`