423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-22 22:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix graph_conv test

Browse Source 
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
This commit is contained in:
Jeremy Reizenstein 2020-08-07 01:57:51 -07:00 committed by Facebook GitHub Bot
parent 7944d24d48
commit 5d9444307f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_graph_conv.py
View File

@ -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()