From 341731642277333c75c181a6e45040218fab8d39 Mon Sep 17 00:00:00 2001
From: Alex Reinking <areinking@adobe.com>
Date: Thu, 29 Aug 2024 14:46:06 -0400
Subject: [PATCH] Allow generating C_SOURCE as an extra output

---
 cmake/HalideGeneratorHelpers.cmake | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cmake/HalideGeneratorHelpers.cmake b/cmake/HalideGeneratorHelpers.cmake
index 237d28f180cd..d953c952e4b1 100644
--- a/cmake/HalideGeneratorHelpers.cmake
+++ b/cmake/HalideGeneratorHelpers.cmake
@@ -240,7 +240,11 @@ function(_Halide_library_from_generator TARGET)
     endif ()
 
     foreach (output IN LISTS ARG_EXTRA_OUTPUTS)
-        _Halide_add_output("${output}" "${TARGET}")
+        if (output STREQUAL "c_source" AND ARG_TYPE STREQUAL "c_source")
+            # Skip this as it was handled above
+        else ()
+            _Halide_add_output("${output}" "${TARGET}")
+        endif ()
     endforeach ()
 
     ## Run the generator
@@ -440,6 +444,7 @@ function(add_halide_library TARGET)
         ASSEMBLY
         BITCODE
         COMPILER_LOG
+        C_SOURCE
         FEATURIZATION
         FUNCTION_INFO_HEADER
         HLPIPE