pytorch
c958dd54 - [TensorPipe] Add guards against transferring GPU tensors (#40167)

Commit
4 years ago
[TensorPipe] Add guards against transferring GPU tensors (#40167) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40167 In v1.6 TensorPipe will not support transferring GPU tensors so, just like other agents, it should raise the appropriate errors when the user attempts to do so. One such error is when sending the arguments, another is when sending the result. ghstack-source-id: 106059723 Test Plan: Re-enabled the test for this Differential Revision: D22091737 fbshipit-source-id: 23dda98bc006333c6179361e8cfaf00ecda06408
Author
lw lw
Parents
Loading