423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-19 05:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

lint fixes

Browse Source 
Summary: Ran `dev/linter.sh`.

Reviewed By: bottler

Differential Revision: D19761062

fbshipit-source-id: 1a49abe4a5f2bc7641b2b46e254aa77e6a48aa7d
This commit is contained in:
Nikhila Ravi
2020-02-13 20:49:18 -08:00
committed by Facebook Github Bot
parent 29cd181a83
commit 97acf16de2
3 changed files with 41 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_nearest_neighbor_points.py
View File

@@ -43,21 +43,21 @@ class TestNearestNeighborPoints(unittest.TestCase):
"""
Test cuda output vs naive python implementation.
"""
device = torch.device('cuda:0')
device = torch.device("cuda:0")
self._test_nn_helper(device)
def test_nn_cpu(self):
"""
Test cpu output vs naive python implementation
"""
device = torch.device('cpu')
device = torch.device("cpu")
self._test_nn_helper(device)
@staticmethod
def bm_nn_points_cpu_with_init(
N: int = 4, D: int = 4, P1: int = 128, P2: int = 128
):
device = torch.device('cpu')
device = torch.device("cpu")
x = torch.randn(N, P1, D, device=device)
y = torch.randn(N, P2, D, device=device)