Searched refs:CUDAStreamForId (Results 1 – 1 of 1) sorted by relevance
257 CUDAStream CUDAStreamForId(DeviceIndex device_index, StreamId stream_id) { in CUDAStreamForId() function337 return CUDAStreamForId(device_index, makeStreamId(id_type, idx)); in getStreamFromPool()350 return CUDAStreamForId(device_index, reinterpret_cast<int64_t>(ext_stream)); in getStreamFromExternal()360 return CUDAStreamForId(device_index, makeStreamId(StreamIdType::DEFAULT, 0)); in getDefaultCUDAStream()370 return CUDAStreamForId(device_index, current_streams[device_index]); in getCurrentCUDAStream()