mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
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
This commit is contained in:
parent
7944d24d48
commit
5d9444307f
@ -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()
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user