feat(CUDA): add new target for testing

We want to try a simplified GPU pipeline for CUDA and test on top of it

Author: chrsmcgrr

compiler/plugins/iree_compiler_plugin.cmake

@@ -17,6 +17,7 @@ if(IREE_INPUT_TOSA)
 endif()
 
 if(IREE_TARGET_BACKEND_CUDA)
+  add_subdirectory(${CMAKE_CURRENT_LIST_DIR}/target/CellarCUDA target/CellarCUDA)
   add_subdirectory(${CMAKE_CURRENT_LIST_DIR}/target/CUDA target/CUDA)
 endif()
 

compiler/plugins/target/CellarCUDA/BUILD.bazel

@@ -0,0 +1,67 @@
+# Copyright 2022 The IREE 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", "iree_compiler_register_plugin")
+
+package(
+    default_visibility = ["//visibility:public"],
+    features = ["layering_check"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+iree_compiler_register_plugin(
+    plugin_id = "hal_target_cellar_cuda",
+    target = ":CellarCUDA",
+)
+
+iree_compiler_cc_library(
+    name = "CellarCUDA",
+    srcs = [
+        "CUDATarget.cpp",
+    ],
+    deps = [
+        "//compiler/plugins/target/CellarCUDA/CodeGen",
+        "//compiler/src/iree/compiler/Codegen",
+        "//compiler/src/iree/compiler/Codegen/Common",
+        "//compiler/src/iree/compiler/Codegen/Dialect/Codegen/IR:IREECodegenDialect",
+        "//compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils:KnownTargets",
+        "//compiler/src/iree/compiler/Codegen/LLVMGPU",
+        "//compiler/src/iree/compiler/Codegen/Utils",
+        "//compiler/src/iree/compiler/Dialect/HAL/Target",
+        "//compiler/src/iree/compiler/Dialect/HAL/Utils:ExecutableDebugInfoUtils",
+        "//compiler/src/iree/compiler/Dialect/HAL/Utils:LLVMLinkerUtils",
+        "//compiler/src/iree/compiler/PluginAPI",
+        "//compiler/src/iree/compiler/Utils",
+        "//runtime/src/iree/base/internal/flatcc:building",
+        "//runtime/src/iree/schemas:cuda_executable_def_c_fbs",
+        "//runtime/src/iree/schemas:executable_debug_info_c_fbs",
+        "@iree_cuda//:libdevice_embedded",
+        "@llvm-project//llvm:Analysis",
+        "@llvm-project//llvm:BitReader",
+        "@llvm-project//llvm:BitWriter",
+        "@llvm-project//llvm:Core",
+        "@llvm-project//llvm:IPO",
+        "@llvm-project//llvm:Linker",
+        "@llvm-project//llvm:MC",
+        "@llvm-project//llvm:NVPTXCodeGen",
+        "@llvm-project//llvm:NVPTXInfo",
+        "@llvm-project//llvm:NVPTXUtilsAndDesc",
+        "@llvm-project//llvm:Passes",
+        "@llvm-project//llvm:Support",
+        "@llvm-project//llvm:Target",
+        "@llvm-project//mlir:BuiltinToLLVMIRTranslation",
+        "@llvm-project//mlir:GPUDialect",
+        "@llvm-project//mlir:LLVMDialect",
+        "@llvm-project//mlir:LLVMToLLVMIRTranslation",
+        "@llvm-project//mlir:NVGPUDialect",
+        "@llvm-project//mlir:NVVMDialect",
+        "@llvm-project//mlir:NVVMToLLVMIRTranslation",
+        "@llvm-project//mlir:Pass",
+        "@llvm-project//mlir:Support",
+        "@llvm-project//mlir:ToLLVMIRTranslation",
+        "@llvm-project//mlir:TransformDialect",
+    ],
+)

compiler/plugins/target/CellarCUDA/CMakeLists.txt

@@ -0,0 +1,69 @@
+################################################################################
+# Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from           #
+# compiler/plugins/target/CellarCUDA/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_compiler_register_plugin(
+  PLUGIN_ID
+    hal_target_cellar_cuda
+  TARGET
+    ::CellarCUDA
+)
+
+iree_cc_library(
+  NAME
+    CellarCUDA
+  SRCS
+    "CUDATarget.cpp"
+  DEPS
+    LLVMAnalysis
+    LLVMBitReader
+    LLVMBitWriter
+    LLVMCore
+    LLVMLinker
+    LLVMMC
+    LLVMNVPTXCodeGen
+    LLVMNVPTXDesc
+    LLVMNVPTXInfo
+    LLVMPasses
+    LLVMSupport
+    LLVMTarget
+    LLVMipo
+    MLIRBuiltinToLLVMIRTranslation
+    MLIRGPUDialect
+    MLIRLLVMDialect
+    MLIRLLVMToLLVMIRTranslation
+    MLIRNVGPUDialect
+    MLIRNVVMDialect
+    MLIRNVVMToLLVMIRTranslation
+    MLIRPass
+    MLIRSupport
+    MLIRTargetLLVMIRExport
+    MLIRTransformDialect
+    iree::base::internal::flatcc::building
+    iree::compiler::Codegen
+    iree::compiler::Codegen::Common
+    iree::compiler::Codegen::Dialect::Codegen::IR::IREECodegenDialect
+    iree::compiler::Codegen::Dialect::GPU::TargetUtils::KnownTargets
+    iree::compiler::Codegen::LLVMGPU
+    iree::compiler::Codegen::Utils
+    iree::compiler::Dialect::HAL::Target
+    iree::compiler::Dialect::HAL::Utils::ExecutableDebugInfoUtils
+    iree::compiler::Dialect::HAL::Utils::LLVMLinkerUtils
+    iree::compiler::PluginAPI
+    iree::compiler::Utils
+    iree::compiler::plugins::target::CellarCUDA::CodeGen
+    iree::schemas::cuda_executable_def_c_fbs
+    iree::schemas::executable_debug_info_c_fbs
+    iree_cuda::libdevice_embedded
+  PUBLIC
+)
+
+### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###