Function getStream

src/backend/cuda/platform.cpp:397–406

}

cudaStream_t getStream(int device) {
    static once_flag streamInitFlags[DeviceManager::MAX_DEVICES];

    call_once(streamInitFlags[device], [device]() {
        DeviceManager &inst = DeviceManager::getInstance();
        CUDA_CHECK(cudaStreamCreate(&(inst.streams[device])));
    });

    return DeviceManager::getInstance().streams[device];
}

cudaStream_t getActiveStream() { return getStream(getActiveDeviceId()); }