提交 470c218f authored 作者: Frederic Bastien's avatar Frederic Bastien

correctly test gpu code. The test was disabled by accident.

上级 42d40e33
......@@ -400,11 +400,16 @@ def test_neibs_grad_verify_grad_warp_centered():
try:
unittest_tools.verify_grad(fn, [images_val], mode=mode_without_gpu)
raise Exception("Expected an error")
if cuda.cuda_available:
unittest_tools.verify_grad(fn, [images_val], mode=mode_with_gpu)
except NotImplementedError:
pass
if cuda.cuda_available:
try:
unittest_tools.verify_grad(fn, [images_val], mode=mode_with_gpu)
raise Exception("Expected an error")
except NotImplementedError:
pass
if __name__ == '__main__':
#test_neibs_gpu()
#test_neibs()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论