Thesis/pytorch-dnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

no gpu on travis

Browse Source
This commit is contained in:
ixaxaar 2017-12-01 11:24:09 +05:30
parent 6e5ffa1770
commit 672ffe130e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_indexes.py
View File

@ -1,5 +1,5 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import pytest import pytest
import numpy as np import numpy as np
@ -36,7 +36,7 @@ def test_indexes():
d = T.ones(n, cell_size) d = T.ones(n, cell_size)
q = T.ones(1, cell_size) q = T.ones(1, cell_size)
for gpu_id in (0, -1): for gpu_id in (-1, -1):
i = Index(cell_size=cell_size, nr_cells=nr_cells, K=K, probes=probes, gpu_id=gpu_id) i = Index(cell_size=cell_size, nr_cells=nr_cells, K=K, probes=probes, gpu_id=gpu_id)
d = d if gpu_id == -1 else d.cuda(gpu_id) d = d if gpu_id == -1 else d.cuda(gpu_id)