Jeremy Reizenstein 7f1e63aed1 Take care with single integers on gpu
Summary:
Pytorch seems to be becoming stricter about integer tensors of shape `(1,)` on GPU and not allowing them to be used as `int`s. For example the following no longer works on pytorch master,
    foo = torch.tensor([3, 5, 3], device="cuda:0")
    torch.arange(10) + foo[0]
because this is the sum of tensors on different devices.

Here fix tests which recently broke because of this.

Reviewed By: nikhilaravi

Differential Revision: D21929745

fbshipit-source-id: 25374f70468d1c895372766f1a9dd61df0833957
2020-06-10 14:13:30 -07:00
..
2020-01-23 11:53:46 -08:00
2020-04-24 09:11:04 -07:00
2020-04-14 17:22:56 -07:00
2020-04-17 07:44:16 -07:00
2020-04-11 00:21:24 -07:00
2020-04-17 07:44:16 -07:00
2020-04-16 14:02:16 -07:00
2020-04-20 12:19:45 -07:00
2020-05-04 09:56:44 -07:00
2020-05-04 09:56:44 -07:00
2020-04-17 07:44:16 -07:00
2020-04-24 09:11:04 -07:00
2020-05-05 11:09:45 -07:00
2020-04-24 09:11:04 -07:00
2020-04-24 09:11:04 -07:00
2020-04-24 09:11:04 -07:00
2020-05-04 09:56:44 -07:00
2020-05-22 22:38:29 -07:00
2020-04-06 06:41:00 -07:00
2020-04-24 09:11:04 -07:00
2020-05-04 09:56:44 -07:00
2020-05-04 09:56:44 -07:00
2020-05-04 09:56:44 -07:00
2020-01-23 11:53:46 -08:00