Method_Meta contains_backend check

Summary: For a runtime that supports models of a couple different types it helps to be able to check ahead of time whether certain backend types are present. In our case, there are processors that can use this information to make scheduling decisions without having the actual required backend implementations present.

Differential Revision: D69143825

Author: cmt0

runtime/executor/method_meta.cpp

@@ -248,5 +248,16 @@ size_t MethodMeta::num_instructions() const {
   return num_instructions;
 }
 
+bool MethodMeta::contains_backend(const char* backend_name) const {
+  const auto delegates = s_plan_->delegates();
+  for (size_t i = 0; i < delegates->size(); i++) {
+    auto delegate = delegates->Get(i);
+    if (strcmp(delegate->id()->c_str(), backend_name) == 0) {
+      return true;
+    }
+  }
+  return false;
+}
+
 } // namespace runtime
 } // namespace executorch

runtime/executor/method_meta.h

@@ -192,6 +192,13 @@ class MethodMeta final {
    */
   ET_EXPERIMENTAL size_t num_instructions() const;
 
+  /**
+   * Check to see if a backend is used in this method.
+   *
+   * @returns Whether a backend is used in this method.
+   */
+  ET_EXPERIMENTAL bool contains_backend(const char* backend_name) const;
+
   /**
    * DEPRECATED: Use num_memory_planned_buffers() instead.
    */