Skip to content
This repository has been archived by the owner on Jun 27, 2024. It is now read-only.

Implement async compiler plugin #159

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

yijia1212
Copy link

This PR implement iree compiler plugin to parse/transform upstream MLIR async dialect:

  • Create async dialect
  • Convert mlir async ops/types to ops/types in iree custom module

@ezhulenev ezhulenev self-requested a review June 22, 2023 03:35
@@ -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)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you please move it under nvgpu folder (and namespace openxla::compiler::nvgpu::async), it's not quite accurate, but it will get it covered by CI jobs, and we'll find a better place later


IREE_DEFINE_COMPILER_OPTION_FLAGS(AsyncOptions);

extern "C" bool iree_register_compiler_plugin_openxla_async(
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And this will go into PluginRegistration shared by other plugins

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants