Commit 97ccef85 authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Fix error in the cuda context.

parent bbfbafa8

graph_framework/cuda_context.hpp

+10 −4
Original line number Diff line number Diff line
@@ -88,7 +88,7 @@ namespace gpu { 
//------------------------------------------------------------------------------

///  @brief Cuda context constructor.

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

        cuda_context() : result_buffer(0) {

        cuda_context() : result_buffer(0), module(0) {

            check_error(cuDeviceGet(&device, 0), "cuDeviceGet");

            check_error(cuDevicePrimaryCtxRetain(&context, device), "cuDevicePrimaryCtxRetain");

            check_error(cuCtxSetCurrent(context), "cuCtxSetCurrent");
@@ -99,13 +99,19 @@ namespace gpu { 
///  @brief Cuda context destructor.

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

        ~cuda_context() {

            if (module) {

                 check_error(cuModuleUnload(module), "cuModuleUnload");

                 module = 0;

            }



            for (auto &[key, value] : kernel_arguments) {

                check_error(cuMemFree(value), "cuMemFree");

            }



            if (result_buffer) {

                check_error(cuMemFree(result_buffer), "cuMemFree");

                result_buffer = 0;

            }



            check_error(cuStreamDestroy(stream), "cuStreamDestroy");

            check_error(cuDevicePrimaryCtxRelease(device), "cuDevicePrimaryCtxRelease");
@@ -332,7 +338,7 @@ namespace gpu { 
                            T *source) {

            size_t size;

            check_error(cuMemGetAddressRange(NULL, &size, kernel_arguments[node.get()]), "cuMemGetAddressRange");

            check_error_async(cuMemcpyHtoDAsync(source, kernel_arguments[node.get()], size, stream), "cuMemcpyHtoDAsync");

            check_error_async(cuMemcpyHtoDAsync(kernel_arguments[node.get()], source, size, stream), "cuMemcpyHtoDAsync");

        }



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