Function c10::xpu::getStreamFromPool(const bool, DeviceIndex)

• Defined in File XPUStream.h

Function Documentation

XPUStream c10::xpu::getStreamFromPool(const bool isHighPriority = false, DeviceIndex device = -1)

Get a stream from the pool in a round-robin fashion.

You can request a stream from the highest priority pool by setting isHighPriority to true for a specific device.