Method set_gpu_id

XM/include/Utils/memory.h:29–32

        DeviceStream(const size_s gpu_id): gpu_id(gpu_id), stream(NULL) {}

        inline void set_gpu_id(const size_s gpu_id) {
            this->gpu_id = gpu_id;
            return;
        }
        inline void activate() {
            CHECK_CUDA( cudaSetDevice(this->gpu_id) );
            CHECK_CUDA( cudaStreamCreateWithFlags(&this->stream, cudaStreamNonBlocking) );