Hi,
I hope to transfor data from cpu to gpu in the current stream, and synchronize the current stream.
So I wonder which stream will be synchronized when ‘.cuda()’ is called? The current stream? or it just act as torch.cuda.synchronize()?
Hi,
I hope to transfor data from cpu to gpu in the current stream, and synchronize the current stream.
So I wonder which stream will be synchronized when ‘.cuda()’ is called? The current stream? or it just act as torch.cuda.synchronize()?