Core ML has added index_put support, no need to skip

yifan_shen3 · yifan_shen3 · commit b9b953bf7f12 · 2024-04-10T14:59:44.000-07:00
diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py
@@ -605,9 +605,6 @@ 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=[
-                    "aten.index_put.default",
-                ],
                 compile_specs=compile_specs,
             )
         )

Original file line number	Diff line number	Diff line change
`@@ -605,9 +605,6 @@ def _export_llama(modelname, args) -> str: # noqa: C901`
`605`	`605`	`partitioners.append(`
`606`	`606`	# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `apple`
`607`	`607`	`CoreMLPartitioner(`
`608`		`- skip_ops_for_coreml_delegation=[`
`609`		`- "aten.index_put.default",`
`610`		`- ],`
`611`	`608`	`compile_specs=compile_specs,`
`612`	`609`	`)`
`613`	`610`	`)`