From 5d9444307f5ccff5fdc15b7b1ed3eaf526c0917e Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Fri, 7 Aug 2020 01:57:51 -0700 Subject: [PATCH] fix graph_conv test Summary: The recently added part of a test was assuming that the random gpu was gpu 0. Reviewed By: nikhilaravi Differential Revision: D22948397 fbshipit-source-id: 88107e19fc3118e763f95be43a614941176a08f9 --- tests/test_graph_conv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_graph_conv.py b/tests/test_graph_conv.py index 75ff4f90..a3ac605d 100644 --- a/tests/test_graph_conv.py +++ b/tests/test_graph_conv.py @@ -113,7 +113,7 @@ class TestGraphConv(TestCaseMixin, unittest.TestCase): neighbor_sums_cpu = gather_scatter(verts_cpu, edges_cpu, False) neighbor_sums = gather_scatter_python(verts, edges, False) randoms = torch.rand_like(neighbor_sums) - (neighbor_sums_cuda * randoms.cuda()).sum().backward() + (neighbor_sums_cuda * randoms.to(device)).sum().backward() (neighbor_sums_cpu * randoms).sum().backward() (neighbor_sums * randoms).sum().backward()