提交 1df8f73e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add comment justifying local imports.

上级 d67b821e
...@@ -1121,6 +1121,7 @@ def local_gpu_conv(node): ...@@ -1121,6 +1121,7 @@ def local_gpu_conv(node):
@local_optimizer([GpuConv]) @local_optimizer([GpuConv])
def local_conv_fft_valid(node): def local_conv_fft_valid(node):
# import locally to avoid pycuda warnings
from theano.sandbox.cuda.fftconv import conv2d_fft from theano.sandbox.cuda.fftconv import conv2d_fft
if (isinstance(node.op, GpuConv) and if (isinstance(node.op, GpuConv) and
node.op.border_mode == 'valid' and node.op.border_mode == 'valid' and
...@@ -1130,6 +1131,7 @@ def local_conv_fft_valid(node): ...@@ -1130,6 +1131,7 @@ def local_conv_fft_valid(node):
@local_optimizer([GpuConv]) @local_optimizer([GpuConv])
def local_conv_fft_full(node): def local_conv_fft_full(node):
# import locally to avoid pycuda warnings
from theano.sandbox.cuda.fftconv import conv2d_fft from theano.sandbox.cuda.fftconv import conv2d_fft
if (isinstance(node.op, GpuConv) and if (isinstance(node.op, GpuConv) and
node.op.border_mode == 'full' and node.op.border_mode == 'full' and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论