Method set_gpu_id

XM/include/Utils/memory.h:57–60

        DeviceBlasHandle(const size_s gpu_id): gpu_id(gpu_id), cublas_handle(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_CUBLAS( cublasCreate_v2(&this->cublas_handle) );