Introduce RegisterAllTransformation API in compiler plugin c api.

This function will be implemented in vendor compiler plugin to register transformations.

LiteRT-PiperOrigin-RevId: 815881905

Author: YunanAZ

litert/vendors/c/BUILD

@@ -28,6 +28,7 @@ cc_library(
         "//litert/c:litert_environment_options",
         "//litert/c:litert_model",
         "//litert/c:litert_options",
+        "//litert/c:litert_rewriter",
     ],
 )
 
@@ -37,10 +38,10 @@ cc_library(
     deps = [
         ":litert_compiler_plugin",
         "//litert/c:litert_common",
-        "//litert/c:litert_environment",
         "//litert/c:litert_environment_options",
         "//litert/c:litert_model",
         "//litert/c:litert_options",
+        "//litert/c:litert_rewriter",
         "@com_google_absl//absl/strings:string_view",
     ],
 )

litert/vendors/c/litert_compiler_plugin.h

@@ -18,9 +18,7 @@
 #include <stddef.h>
 
 #include "litert/c/litert_common.h"
-#include "litert/c/litert_environment_options.h"
-#include "litert/c/litert_model.h"
-#include "litert/c/litert_options.h"
+#include "litert/c/litert_rewriter.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -111,6 +109,16 @@ LITERT_CAPI_EXPORT LiteRtStatus LiteRtGetCompiledResultCallInfo(
 LITERT_CAPI_EXPORT LiteRtStatus LiteRtGetNumCompiledResultCalls(
     LiteRtCompiledResult compiled_result, LiteRtParamIndex* num_calls);
 
+// Allow the compiler plugin to registers all the graph transformations
+// required. This function populates the provided arrays with pattern functions
+// and their corresponding names. Registered patterns will be applied to the
+// graph before partition and compilation.
+//
+// Experimental: Unstable ABI, function signature is subject to change.
+LiteRtStatus LiteRtCompilerPluginRegisterAllTransformations(
+    LiteRtCompilerPlugin compiler_plugin, LiteRtPatternFn** pattern_fns,
+    const char*** transformation_names, LiteRtParamIndex* num_patterns);
+
 #ifdef __cplusplus
 }
 #endif  // __cplusplus

litert/vendors/c/litert_compiler_plugin_api.h

@@ -18,6 +18,7 @@
 #include <stddef.h>
 
 #include "litert/c/litert_common.h"
+#include "litert/c/litert_rewriter.h"
 #include "litert/vendors/c/litert_compiler_plugin.h"
 
 // Wrapper for dynamically loaded LiteRtCompilerPlugin library. See
@@ -75,6 +76,10 @@ typedef LiteRtStatus (*LiteRtGetCompiledResultCallInfoT)(
 typedef LiteRtStatus (*LiteRtGetNumCompiledResultCallsT)(
     LiteRtCompiledResult, LiteRtParamIndex* num_calls);
 
+typedef LiteRtStatus (*LiteRtCompilerPluginRegisterAllTransformationsT)(
+    LiteRtCompilerPlugin compiler_plugin, LiteRtPatternFn** pattern_fns,
+    const char*** transformation_names, LiteRtParamIndex* num_patterns);
+
 //
 // Function Pointer Container
 //
@@ -101,6 +106,7 @@ struct LiteRtCompilerPluginApi {
   LiteRtCompiledResultNumByteCodeModulesT get_compiled_result_num_byte_code;
   LiteRtGetCompiledResultCallInfoT get_compiled_result_call_info;
   LiteRtGetNumCompiledResultCallsT get_compiled_result_num_calls;
+  LiteRtCompilerPluginRegisterAllTransformationsT register_all_transformations;
 };
 
 #ifdef __cplusplus
@@ -142,6 +148,9 @@ static constexpr absl::string_view kLiteRtGetCompiledResultCallInfo =
     "LiteRtGetCompiledResultCallInfo";
 static constexpr absl::string_view kLiteRtGetNumCompiledResultCalls =
     "LiteRtGetNumCompiledResultCalls";
+static constexpr absl::string_view
+    kLiteRtCompilerPluginRegisterAllTransformations =
+        "LiteRtCompilerPluginRegisterAllTransformations";
 
 #endif  // __cplusplus