Kehan Wang

icon indicating a website link https://jason-khan.github.io/ icon indicating an email [email protected]

icon location Berkeley, CA icon location Googler. Previously Master student @ UC Berkeley interested in Computer Vision and Representation Learning

Results 1 issues of Kehan Wang

Support GPU tensor transport on pytorch

1 comment

According to https://pytorch.org/docs/stable/rpc.html, tensorpipe currently does not support sending GPU tensors over the wire. I'm wondering when this functionality would be implemented and integrated with pytorch.