[MPS Delegate] Add new mps runtime with support for FP16, lifted and unlifted graphs (iOS15+, macOS12+)

backends/apple/mps/CMakeLists.txt

@@ -20,14 +20,46 @@ if(NOT EXECUTORCH_ROOT)
   set(EXECUTORCH_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../../..)
 endif()
 
+if(NOT FLATC_EXECUTABLE)
+  set(FLATC_EXECUTABLE flatc)
+endif()
+
 set(_common_compile_options -Wno-deprecated-declarations)
 set(_common_include_directories ${EXECUTORCH_ROOT}/..)
 
+set(_mps_schema__include_dir "${CMAKE_BINARY_DIR}/schema/include")
+
+# Paths to headers generated from the .fbs files.
+set(_mps_schema__outputs)
+foreach(fbs_file ${_mps_schema__srcs})
+  string(REGEX REPLACE "serialization/([^/]+)[.]fbs$" "\\1_generated.h"
+                       generated "${fbs_file}")
+  list(APPEND _mps_schema__outputs
+       "${_mps_schema__include_dir}/executorch/${generated}")
+endforeach()
+
+# Generate the headers from the .fbs files.
+add_custom_command(
+  OUTPUT ${_mps_schema__outputs}
+  COMMAND
+    ${FLATC_EXECUTABLE} --cpp --cpp-std c++11 --scoped-enums -o
+    "${_mps_schema__include_dir}/executorch/backends/apple/mps"
+    ${_mps_schema__srcs}
+  WORKING_DIRECTORY ${EXECUTORCH_ROOT}
+  COMMENT "Generating mps_schema headers"
+  VERBATIM)
+
+add_library(mps_schema INTERFACE ${_mps_schema__outputs})
+set_target_properties(mps_schema PROPERTIES LINKER_LANGUAGE CXX)
+target_include_directories(
+mps_schema INTERFACE ${_mps_schema__include_dir}
+                     ${EXECUTORCH_ROOT}/third-party/flatbuffers/include
+                     ${_common_include_directories})
+
 list(TRANSFORM _mps_backend__srcs PREPEND "${EXECUTORCH_ROOT}/")
 add_library(mpsdelegate ${_mps_backend__srcs})
-target_link_libraries(mpsdelegate PRIVATE baundled_program
+target_link_libraries(mpsdelegate PRIVATE baundled_program mps_schema
                                           ${_executor_runner_libs})
-target_include_directories(mpsdelegate PUBLIC ${_common_include_directories})
 
 install(
   TARGETS mpsdelegate