Rate this Page
★★★★★

torch.cuda.set_stream#

torch.cuda.set_stream(stream)[source]#
Set the current stream.This is a wrapper API to set the stream.

Usage of this function is discouraged in favor of thestreamcontext manager.

Parameters

stream (Stream) – selected stream. This function is a no-opif this argument isNone.