pfnet/pytorch-pfn-extras

Pass `device_id` to `cupy.cuda.ExternalStream` from `torch.cuda.Stream.device`

Closed this issue a year ago · 0 comments

kmaehashi commented a year ago

Pass `device_id` to `cupy.cuda.ExternalStream` from `torch.cuda.Stream.device`