提交 c6129572 authored 作者: Frederic's avatar Frederic

Fix import of conv3d2d when cuda isn't available.

上级 ebb78f6b
...@@ -286,5 +286,6 @@ def make_gpu_optimizer(op, to_gpu): ...@@ -286,5 +286,6 @@ def make_gpu_optimizer(op, to_gpu):
local_to_gpu.__name__ = "local_to_gpu_" + op.__name__ local_to_gpu.__name__ = "local_to_gpu_" + op.__name__
cuda.opt.register_opt()(local_to_gpu) cuda.opt.register_opt()(local_to_gpu)
make_gpu_optimizer(DiagonalSubtensor, [0]) if cuda.cuda_available:
make_gpu_optimizer(IncDiagonalSubtensor, [0, 3]) make_gpu_optimizer(DiagonalSubtensor, [0])
make_gpu_optimizer(IncDiagonalSubtensor, [0, 3])
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论