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

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

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