Allow generating C_SOURCE as an extra output

halide · Aug 29, 2024 · 3417316 · 3417316
1 parent d72df59
commit 3417316
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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