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

@@ -139,6 +139,10 @@ def mps(cls) -> bool:
     def xnnpack(cls) -> bool:
         return cls._is_cmake_arg_enabled("EXECUTORCH_BUILD_XNNPACK", 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_cmake_arg_enabled(
@@ -843,6 +847,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,