commit 77035d151e08f7aee97689cf1f29bbf8a968a61e
author Soumith Chintala <soumith@gmail.com> Fri Apr 28 07:30:25 2017 -0400
committer Soumith Chintala <soumith@gmail.com> Fri Apr 28 07:30:25 2017 -0400
tree 488c040a1bbef69e7ece679794ad81f20a311be3
parent 50c9c23525ccf0b060b181b4466f549c5b0676dd

make topk test unique

test/test_cuda.py

diff --git a/test/test_cuda.py b/test/test_cuda.py
index d00fb55..cad28e0 100644
--- a/test/test_cuda.py
+++ b/test/test_cuda.py
@@ -257,9 +257,9 @@
     ('transpose', new_t(1, 2, 3, 4), lambda t: [1, 2],),
     ('transpose', new_t(1, 2, 3, 4), lambda t: [-1, -2], 'neg_dim'),
     ('to_list', small_3d, lambda t: [],),
-    ('topk', small_3d, lambda t: [2, 1, False, True], 'dim_sort'),
-    ('topk', small_3d, lambda t: [2, -1, False, True], 'neg_dim_sort'),
-    ('topk', small_3d, lambda t: [2, 1, True, True], 'dim_desc_sort'),
+    ('topk', small_3d_unique, lambda t: [2, 1, False, True], 'dim_sort'),
+    ('topk', small_3d_unique, lambda t: [2, -1, False, True], 'neg_dim_sort'),
+    ('topk', small_3d_unique, lambda t: [2, 1, True, True], 'dim_desc_sort'),
     ('trace', medium_2d, lambda t: [],),
     ('tril', medium_2d, lambda t: [],),
     ('tril', medium_2d, lambda t: [2], 'positive'),