update usage

Author: jathu

backends/apple/coreml/compiler/coreml_preprocess.py

@@ -16,8 +16,8 @@
 
 import coremltools as ct
 import coremltools.optimize as cto
-import executorchcoreml
 
+from executorch.backends.apple.coreml import inmemoryfs
 from executorch.exir.backend.backend_details import (
     BackendDetails,
     ExportedProgram,
@@ -385,8 +385,7 @@ def preprocess_model(
             )
 
         processed_bytes: bytes = (
-            executorchcoreml.flatten_directory_contents(str(model_dir_path.resolve()))
-            or b""
+            inmemoryfs.flatten_directory_contents(str(model_dir_path.resolve())) or b""
         )
 
         debug_handle_map: Optional[Dict[str, Tuple[int]]] = None

backends/apple/coreml/executorchcoreml.pyi renamed to backends/apple/coreml/inmemoryfs.pyi

@@ -62,8 +62,6 @@ if [ $STATUS -ne 0 ]; then
     exit 1
 fi
 
-sh "$COREML_DIR_PATH/scripts/install_inmemoryfs.sh"
-
 echo "${green}ExecuTorch: Copying protobuf files."
 mkdir -p "$COREML_DIR_PATH/runtime/sdk/format/"
 cp -rf "$PROTOBUF_FILES_DIR_PATH" "$COREML_DIR_PATH/runtime/sdk/format/"

backends/apple/coreml/runtime/inmemoryfs/setup.py

@@ -10,12 +10,9 @@
 
 from typing import List, Optional
 
-import executorchcoreml
-
+from executorch.backends.apple.coreml import inmemoryfs
 from executorch.backends.apple.coreml.compiler import CoreMLBackend
-
 from executorch.exir._serialize._program import deserialize_pte_binary
-
 from executorch.exir.schema import (
     BackendDelegate,
     BackendDelegateDataReference,
@@ -50,7 +47,7 @@ def extract_coreml_models(pte_data: bytes):
             shutil.rmtree(model_path.absolute())
         os.makedirs(model_path.absolute())
 
-        if executorchcoreml.unflatten_directory_contents(
+        if inmemoryfs.unflatten_directory_contents(
             coreml_processed_bytes, str(model_path.absolute())
         ):
             print(f"Core ML models are extracted and saved to path = {model_path}")