Implement async compiler plugin

compiler/src/CMakeLists.txt

@@ -9,3 +9,4 @@ set(IREE_PACKAGE_ROOT_PREFIX "")
 set(IREE_COMPILER_TABLEGEN_INCLUDE_DIRS "${CMAKE_CURRENT_SOURCE_DIR}")
 
 add_subdirectory(openxla/compiler/nvgpu)
+add_subdirectory(openxla/compiler/async)

compiler/src/openxla/compiler/async/BUILD.bazel

@@ -0,0 +1,38 @@
+# Copyright 2023 The OpenXLA Authors
+#
+# Licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+load("@iree_core//build_tools/bazel:build_defs.oss.bzl", "iree_compiler_register_plugin")
+
+package(
+    default_visibility = ["//visibility:public"],
+    features = ["layering_check"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+cc_library(
+    name = "defs",
+    includes = ["../../.."],
+)
+
+cc_library(
+    name = "registration",
+    srcs = [
+        "PluginRegistration.cpp",
+    ],
+    deps = [
+        ":defs",
+        "//compiler/src/openxla/compiler/async/Dialect/Async/IR",
+        "//compiler/src/openxla/compiler/async/Dialect/Async/Transforms",
+        "@iree_core//compiler/src/iree/compiler/PluginAPI",
+        "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Pass",
+    ],
+)
+
+iree_compiler_register_plugin(
+    plugin_id = "openxla_async",
+    target = ":registration",
+)

compiler/src/openxla/compiler/async/CMakeLists.txt

@@ -0,0 +1,44 @@
+################################################################################
+# Autogenerated by ../iree/build_tools/bazel_to_cmake/bazel_to_cmake.py from   #
+# compiler/src/openxla/compiler/async/BUILD.bazel                              #
+#                                                                              #
+# Use iree_cmake_extra_content from iree/build_defs.oss.bzl to add arbitrary   #
+# CMake-only content.                                                          #
+#                                                                              #
+# To disable autogeneration for this file entirely, delete this header.        #
+################################################################################
+
+iree_add_all_subdirs()
+
+iree_cc_library(
+  NAME
+    defs
+  INCLUDES
+    "$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/../../..>"
+    "$<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/../../..>"
+  PUBLIC
+)
+
+iree_cc_library(
+  NAME
+    registration
+  SRCS
+    "PluginRegistration.cpp"
+  DEPS
+    ::defs
+    MLIRIR
+    MLIRPass
+    iree::compiler::PluginAPI
+    openxla::compiler::async::Dialect::Async::IR
+    openxla::compiler::async::Dialect::Async::Transforms
+  PUBLIC
+)
+
+iree_compiler_register_plugin(
+  PLUGIN_ID
+    openxla_async
+  TARGET
+    ::registration
+)
+
+### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###

compiler/src/openxla/compiler/async/Dialect/Async/BUILD.bazel

@@ -0,0 +1,11 @@
+# Copyright 2023 The OpenXLA Authors
+#
+# Licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+package(
+    default_visibility = ["//visibility:public"],
+    features = ["layering_check"],
+    licenses = ["notice"],  # Apache 2.0
+)

compiler/src/openxla/compiler/async/Dialect/Async/CMakeLists.txt

@@ -0,0 +1,13 @@
+################################################################################
+# Autogenerated by ../iree/build_tools/bazel_to_cmake/bazel_to_cmake.py from   #
+# compiler/src/openxla/compiler/async/Dialect/Async/BUILD.bazel                #
+#                                                                              #
+# Use iree_cmake_extra_content from iree/build_defs.oss.bzl to add arbitrary   #
+# CMake-only content.                                                          #
+#                                                                              #
+# To disable autogeneration for this file entirely, delete this header.        #
+################################################################################
+
+iree_add_all_subdirs()
+
+### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###

compiler/src/openxla/compiler/async/Dialect/Async/Conversion/BUILD.bazel

@@ -0,0 +1,28 @@
+# Copyright 2023 The OpenXLA Authors
+#
+# Licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+load("//build_tools/bazel:build_defs.oss.bzl", "iree_compiler_cc_library")
+
+package(
+    default_visibility = ["//visibility:public"],
+    features = ["layering_check"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+cc_library(
+    name = "AsyncToRuntime",
+    srcs = ["ConvertAsyncToRuntime.cpp"],
+    hdrs = ["ConvertAsyncToRuntime.h"],
+    deps = [
+        "//compiler/src/openxla/compiler/async/Dialect/Async/IR",
+        "@iree_core//compiler/src/iree/compiler/Dialect/Util/IR",
+        "@llvm-project//llvm:Support",
+        "@llvm-project//mlir:FuncDialect",
+        "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Pass",
+        "@llvm-project//mlir:Transforms",
+    ],
+)

compiler/src/openxla/compiler/async/Dialect/Async/Conversion/CMakeLists.txt

@@ -0,0 +1,31 @@
+################################################################################
+# Autogenerated by ../iree/build_tools/bazel_to_cmake/bazel_to_cmake.py from   #
+# compiler/src/openxla/compiler/async/Dialect/Async/Conversion/BUILD.bazel     #
+#                                                                              #
+# Use iree_cmake_extra_content from iree/build_defs.oss.bzl to add arbitrary   #
+# CMake-only content.                                                          #
+#                                                                              #
+# To disable autogeneration for this file entirely, delete this header.        #
+################################################################################
+
+iree_add_all_subdirs()
+
+iree_cc_library(
+  NAME
+    AsyncToRuntime
+  HDRS
+    "ConvertAsyncToRuntime.h"
+  SRCS
+    "ConvertAsyncToRuntime.cpp"
+  DEPS
+    LLVMSupport
+    MLIRFuncDialect
+    MLIRIR
+    MLIRPass
+    MLIRTransforms
+    iree::compiler::Dialect::Util::IR
+    openxla::compiler::async::Dialect::Async::IR
+  PUBLIC
+)
+
+### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###