build inmemoryfs when coreml or pybindings enabled

Author: jathu

CMakeLists.txt

@@ -811,6 +811,7 @@ if(EXECUTORCH_BUILD_PYBIND)
 
   if(EXECUTORCH_BUILD_COREML)
     list(APPEND _dep_libs coremldelegate)
+    list(APPEND _dep_libs coreml_inmemoryfs_pybinding)
   endif()
 
   if(EXECUTORCH_BUILD_MPS)
@@ -935,6 +936,7 @@ if(EXECUTORCH_BUILD_EXECUTOR_RUNNER)
 
   if(EXECUTORCH_BUILD_COREML)
     list(APPEND _executor_runner_libs coremldelegate)
+    list(APPEND _executor_runner_libs coreml_inmemoryfs_pybinding)
   endif()
 
   add_executable(executor_runner ${_executor_runner__srcs})

setup.py

@@ -85,6 +85,10 @@ def _is_env_enabled(env_var: str, default: bool = False) -> bool:
     def pybindings(cls) -> bool:
         return cls._is_env_enabled("EXECUTORCH_BUILD_PYBIND", default=False)
 
+    @classmethod
+    def coreml(cls) -> bool:
+        return cls._is_env_enabled("EXECUTORCH_BUILD_COREML", default=False)
+
     @classmethod
     def training(cls) -> bool:
         return cls._is_env_enabled("EXECUTORCH_BUILD_TRAINING", default=False)
@@ -782,6 +786,15 @@ def get_ext_modules() -> List[Extension]:
             ]
         )
 
+    if ShouldBuild.pybindings() or ShouldBuild.coreml():
+        ext_modules.append(
+            BuiltExtension(
+                src="coreml_inmemoryfs_pybinding.*",
+                src_dir="backends/apple/coreml",
+                modpath="executorch.backends.apple.coreml.inmemoryfs",
+            )
+        )
+
     if ShouldBuild.pybindings():
         ext_modules.append(
             # Install the prebuilt pybindings extension wrapper for the runtime,