提交 b18629bd authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Test img2neibs with only float32 on GPU, other dtypes on CPU

上级 aa9590f9
...@@ -20,6 +20,7 @@ else: ...@@ -20,6 +20,7 @@ else:
class T_GpuImages2Neibs(theano.sandbox.test_neighbours.T_Images2Neibs): class T_GpuImages2Neibs(theano.sandbox.test_neighbours.T_Images2Neibs):
mode = mode_with_gpu mode = mode_with_gpu
op = GpuImages2Neibs op = GpuImages2Neibs
dtypes = ['float32']
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论