Commit 9f033528 authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Only include complex headers when compiling a complex kernel.

parent 940ba642

graph_framework/cuda_context.hpp

+4 −2
Original line number Diff line number Diff line
@@ -407,9 +407,11 @@ namespace gpu { 
///  @params[in,out] source_buffer Source buffer stream.

//------------------------------------------------------------------------------

        void create_header(std::ostringstream &source_buffer) {

            if constexpr (jit::is_complex<T> ()) {

                source_buffer << "#include <cuda/std/complex>" << std::endl;

                source_buffer << "#include <special_functions.hpp>" << std::endl;

            }

        }



//------------------------------------------------------------------------------

///  @brief Create kernel prefix.