提交 84b54a57 authored 作者: Frederic's avatar Frederic

Remove useless warning in debugmode.

上级 5f5f63e1
......@@ -1151,6 +1151,13 @@ class _VariableEquivalenceTracker(object):
for e in self.equiv[key]:
print ' ', e
#List of default version of make thunk.
#This is needed to know if the user overrided it.
#The GpuOp will be added here when theano.sandbox.cuda is imported.
default_make_thunk = [theano.gof.Op.make_thunk.im_func]
class _Linker(gof.link.LocalLinker):
"""Special debugging linker"""
def __init__(self, maker):
......@@ -1239,7 +1246,7 @@ class _Linker(gof.link.LocalLinker):
thunks_py.append(None)
# If the op define its own make_thunk, check it
if node.op.make_thunk.im_func != theano.gof.Op.make_thunk.im_func:
if node.op.make_thunk.im_func not in default_make_thunk:
compute_map = {}
for k in node.inputs:
compute_map[k] = [True]
......
......@@ -179,6 +179,7 @@ class GpuOp(theano.gof.Op):
return super(GpuOp, self).make_thunk(node, storage_map,
compute_map, no_recycling)
theano.compile.debugmode.default_make_thunk.append(GpuOp.make_thunk.im_func)
# We must do those import to be able to create the full doc when
# nvcc is not available
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论