提交 b0949585 authored 作者: notoraptor's avatar notoraptor

Make sure tests pass on a GTX 750 Ti GPU.

上级 94e09a7e
...@@ -383,7 +383,7 @@ class Test_TopK(unittest.TestCase): ...@@ -383,7 +383,7 @@ class Test_TopK(unittest.TestCase):
utt.assert_allclose(np.sort(xval[yval]), np.sort(xval[goal])) utt.assert_allclose(np.sort(xval[yval]), np.sort(xval[goal]))
@utt.parameterized.expand(product( @utt.parameterized.expand(product(
((17, 15), (2, 3, 5, 7, 11), (2017, 5, 3)), ((17, 15), (2, 3, 5, 7, 11), (500, 5, 3)), # NB: Test may fail with bigger sizes (e.g. (2017, 5, 3)) due to "too many resources requested" kernel error on some GPUs.
(-1, '(1+n)//2', '-n', '1-n'), (-1, '(1+n)//2', '-n', '1-n'),
('float32', 'int32'), ('float32', 'int32'),
(False,), (False,),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论