Commit 02e2256f authored by Cianciosa, Mark's avatar Cianciosa, Mark
Browse files

Fix cuda device selection index.

parent 35615635

graph_framework/cuda_context.hpp

+1 −1
Original line number Diff line number Diff line
@@ -100,7 +100,7 @@ namespace gpu { 
///

///  @params[in] index Concurrent index.

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

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

        cuda_context(const size_t index) : result_buffer(0), module(0) {

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

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

            check_error(cuCtxSetCurrent(context), "cuCtxSetCurrent");