From e3fbcee3948fb5ed435ef320a8db85d2fe1ac90e Mon Sep 17 00:00:00 2001
From: Dante Gama Dessavre <dante.gamadessavre@gmail.com>
Date: Tue, 14 Sep 2021 18:07:41 -0500
Subject: [PATCH] FIX Add thrust_create_target to install export

---
 cpp/CMakeLists.txt | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
index ffaefc92f4..b91af8a62e 100644
--- a/cpp/CMakeLists.txt
+++ b/cpp/CMakeLists.txt
@@ -456,6 +456,11 @@ Provide targets for cuML.
 cuML is a suite of libraries that implement machine learning algorithms and mathematical primitives
 functions that share compatible APIs with other RAPIDS projects.
 
+]=])
+
+set(code_string
+[=[
+thrust_create_target(cuml::Thrust FROM_OPTIONS)
 ]=])
 
  rapids_export(INSTALL cuml
@@ -463,11 +468,12 @@ functions that share compatible APIs with other RAPIDS projects.
     GLOBAL_TARGETS cuml
     NAMESPACE cuml::
     DOCUMENTATION doc_string
+    FINAL_CODE_BLOCK code_string
     )
 
 ################################################################################################
 # - build export -------------------------------------------------------------------------------
-set(code_string [=[thrust_create_target(cuml::Thrust FROM_OPTIONS)]=])
+
 rapids_export(BUILD cuml
     EXPORT_SET cuml-exports
     GLOBAL_TARGETS cuml