From 9494bd84df3c5b496fc087285af9ff40d7859b6a Mon Sep 17 00:00:00 2001
From: Jack Wolfard <jack@rivosinc.com>
Date: Sat, 15 Jul 2023 02:44:42 -0700
Subject: [PATCH] [mlir][python] Add install target for MLIR Python sources.

Differential Revision: https://reviews.llvm.org/D155362
---
 mlir/cmake/modules/AddMLIRPython.cmake |  7 ++++---
 mlir/python/CMakeLists.txt             | 13 +++++++++++++
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/mlir/cmake/modules/AddMLIRPython.cmake b/mlir/cmake/modules/AddMLIRPython.cmake
index 382f7c0..5349bb8 100644
--- a/mlir/cmake/modules/AddMLIRPython.cmake
+++ b/mlir/cmake/modules/AddMLIRPython.cmake
@@ -174,12 +174,13 @@ function(_mlir_python_install_sources name source_root_dir destination)
     install(
       FILES "${source_root_dir}/${source_relative_path}"
       DESTINATION "${destination}/${dest_relative_dir}"
-      COMPONENT "${name}"
+      COMPONENT mlir-python-sources
     )
   endforeach()
-  get_target_export_arg(${name} MLIR export_to_mlirtargets UMBRELLA mlir-libraries)
+  get_target_export_arg(${name} MLIR export_to_mlirtargets
+    UMBRELLA mlir-python-sources)
   install(TARGETS ${name}
-    COMPONENT ${name}
+    COMPONENT mlir-python-sources
     ${export_to_mlirtargets}
   )
 endfunction()
diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt
index 29152b5..22a55db 100644
--- a/mlir/python/CMakeLists.txt
+++ b/mlir/python/CMakeLists.txt
@@ -518,6 +518,19 @@ add_mlir_python_common_capi_library(MLIRPythonCAPI
 )
 
 ################################################################################
+# Custom targets.
+################################################################################
+
+_flatten_mlir_python_targets(mlir_python_sources_deps MLIRPythonSources)
+add_custom_target("mlir-python-sources" DEPENDS ${mlir_python_sources_deps})
+if(NOT LLVM_ENABLE_IDE)
+  add_llvm_install_targets(install-mlir-python-sources
+    DEPENDS mlir-python-sources
+    COMPONENT mlir-python-sources
+  )
+endif()
+
+################################################################################
 # The fully assembled package of modules.
 # This must come last.
 ################################################################################
-- 
2.7.4