From b9b953bf7f12380f99d9247c2e21070b1bf1a488 Mon Sep 17 00:00:00 2001 From: yifan_shen3 Date: Wed, 10 Apr 2024 14:59:44 -0700 Subject: [PATCH] Core ML has added index_put support, no need to skip --- examples/models/llama2/export_llama_lib.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py index de1e711a2c9..9dedbf47795 100644 --- 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, ) )