Commit fbb11cc1 authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Modify special functions to compile for cuda kernels.

parent 4e406145

graph_framework/cuda_context.hpp

+4 −2
Original line number Diff line number Diff line
@@ -203,12 +203,13 @@ namespace gpu { 
            }



            const std::string temp = arch.str();

            std::array<const char *, 5> options({

            std::array<const char *, 6> options({

                temp.c_str(),

                "--std=c++17",

                "--include-path=" CUDA_INCLUDE,

                "--include-path=" HEADER_DIR,

                "--extra-device-vectorization"

                "--extra-device-vectorization",

		"--device-as-default-execution-space"

            });



            if (nvrtcCompileProgram(kernel_program, options.size(), options.data())) {
@@ -424,6 +425,7 @@ namespace gpu { 
        void create_header(std::ostringstream &source_buffer) {

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

                source_buffer << "#define CUDA_DEVICE_CODE" << std::endl;

		source_buffer << "#define M_PI " << M_PI << std::endl;

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

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

            }

graph_framework/special_functions.hpp

+128 −109

File changed.

Preview size limit exceeded, changes collapsed.

graph_driver/xrays.cpp

+2 −2

File changed.

Contains only whitespace changes.