Skip to content

Commit

Permalink
skip one op for coreml partitioner (#2826)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #2826

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

Reviewed By: mergennachin

Differential Revision: D55680297

fbshipit-source-id: 99225a22b0369583c36fe67709b66b0784a7765d
  • Loading branch information
cccclai authored and facebook-github-bot committed Apr 4, 2024
1 parent fbf834a commit dd06b7a
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion examples/models/llama2/export_llama_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand Down

0 comments on commit dd06b7a

Please sign in to comment.