fix test on cuda

Brennan Vincent · Brennan Vincent · commit 99e51592ec75 · 2019-05-31T14:07:45.000-04:00
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -9058,7 +9058,7 @@ def gen_nontrivial_input(num_src, dtype, device):
                     if TEST_NUMPY:
                         tup1 = torch.nonzero(tensor, as_tuple=True)
                         tup2 = tensor.nonzero(as_tuple=True)
-                        np1 = tensor.numpy().nonzero()
+                        np1 = tensor.cpu().numpy().nonzero()
                         for t in (tup1, tup2):
                             self.assertEqual(len(t), len(np1))
                             for i in range(len(t)):
