Commit 03bb14fc authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Merge branch 'flow_control' into 'main' 

Fix Cuda builds resulting from copy and past error.

See merge request !98
parents 2b6101b7 5d6669e2

graph_framework/cuda_context.hpp

+1 −1
Original line number Diff line number Diff line
@@ -324,7 +324,7 @@ namespace gpu { 
            check_error(cuModuleGetFunction(&function, module, kernel_name.c_str()), "cuModuleGetFunction");



            std::vector<void *> buffers;

            std::set<graph::leaf_node<float, SAFE_MATH> *> needed_buffers;

            std::set<graph::leaf_node<T, SAFE_MATH> *> needed_buffers;



            const size_t buffer_element_size = sizeof(T);

            for (auto &input : inputs) {