fix internal index.Tensor test on wrong device

bdhirsh · facebook-github-bot · commit f4dd151037e4 · 2022-06-29T09:30:37.000-07:00
Summary: After landing pytorch/pytorch#69607, that made it an error to use indexing with `cpu_tensor[cuda_indices]`. There was one outstanding test in fbcode that incorrectly used indexing in that way, which is fixed here Reviewed By: bottler, osalpekar Differential Revision: D37128838 fbshipit-source-id: 611b6f717b5b5d89fa61fd9ebeb513ad7e65a656
diff --git a/tests/test_texturing.py b/tests/test_texturing.py
@@ -131,7 +131,7 @@ def test_faces_verts_textures(self):
         )
 
         # define TexturesVertex
-        verts_texture = torch.rand(verts.shape)
+        verts_texture = torch.rand(verts.shape, device=device)
         textures = TexturesVertex(verts_features=verts_texture)
 
         # compute packed faces

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def test_faces_verts_textures(self):`
`131`	`131`	`)`
`132`	`132`
`133`	`133`	`# define TexturesVertex`
`134`		`- verts_texture = torch.rand(verts.shape)`
	`134`	`+ verts_texture = torch.rand(verts.shape, device=device)`
`135`	`135`	`textures = TexturesVertex(verts_features=verts_texture)`
`136`	`136`
`137`	`137`	`# compute packed faces`