skip two ops for coreml partitioner (#2826)

cccclai · facebook-github-bot · commit f1ef94b26db8 · 2024-04-02T22:54:17.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
```

Differential Revision: D55680297
diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py
@@ -714,7 +714,7 @@ 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.tensor", "aten.index_put.default"], compile_specs=compile_specs
             )
         )
         modelname = f"coreml_{modelname}"

Original file line number	Diff line number	Diff line change
`@@ -714,7 +714,7 @@ def _export_llama(modelname, args) -> str: # noqa: C901`
`714`	`714`	`partitioners.append(`
`715`	`715`	# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `apple`
`716`	`716`	`CoreMLPartitioner(`
`717`		`- skip_ops_for_coreml_delegation=None, compile_specs=compile_specs`
	`717`	`+ skip_ops_for_coreml_delegation=["aten.index.tensor", "aten.index_put.default"], compile_specs=compile_specs`
`718`	`718`	`)`
`719`	`719`	`)`
`720`	`720`	`modelname = f"coreml_{modelname}"`