提交 6768002c authored 作者: Frederic's avatar Frederic

Better error msg

上级 47be7bd3
...@@ -144,8 +144,11 @@ def raise_with_op(node, thunk=None, exc_info=None, storage_map=None): ...@@ -144,8 +144,11 @@ def raise_with_op(node, thunk=None, exc_info=None, storage_map=None):
clients = [[c[0] for c in var.clients] for var in node.outputs] clients = [[c[0] for c in var.clients] for var in node.outputs]
detailed_err_msg += ("Inputs shapes: %s" % shapes + detailed_err_msg += ("Inputs shapes: %s" % shapes +
"\nInputs strides: %s" % strides + "\nInputs strides: %s" % strides +
"\nInputs values: %s" % scalar_values + "\nInputs values: %s" % scalar_values)
"\nOutputs clients: %s\n" % clients) if hasattr(node.op, '__input_name__'):
detailed_err_msg += "\nInputs name: %s\n" % str(node.op.__input_name__)
detailed_err_msg += "\nOutputs clients: %s\n" % clients
else: else:
hints.append( hints.append(
"HINT: Use another linker then the c linker to" "HINT: Use another linker then the c linker to"
......
...@@ -377,6 +377,8 @@ class GpuDnnConv(DnnBase, COp): ...@@ -377,6 +377,8 @@ class GpuDnnConv(DnnBase, COp):
:param descr: the convolution descriptor :param descr: the convolution descriptor
""" """
__props__ = ('workmem', 'inplace') __props__ = ('workmem', 'inplace')
__input_name__ = ('image', 'kernel', 'output',
'descriptor', 'alpha', 'beta')
def __init__(self, workmem=None, inplace=False): def __init__(self, workmem=None, inplace=False):
""" """
...@@ -501,6 +503,7 @@ class GpuDnnConvGradW(DnnBase, COp): ...@@ -501,6 +503,7 @@ class GpuDnnConvGradW(DnnBase, COp):
""" """
__props__ = ('inplace',) __props__ = ('inplace',)
__input_name__ = ('image', 'grad', 'output', 'descriptor', 'alpha', 'beta')
def __init__(self, inplace=False): def __init__(self, inplace=False):
COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_gw.c"], COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_gw.c"],
...@@ -573,6 +576,8 @@ class GpuDnnConvGradI(DnnBase, COp): ...@@ -573,6 +576,8 @@ class GpuDnnConvGradI(DnnBase, COp):
""" """
__props__ = ('inplace',) __props__ = ('inplace',)
__input_name__ = ('kernel', 'grad', 'output',
'descriptor', 'alpha', 'beta')
def __init__(self, inplace=False): def __init__(self, inplace=False):
COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_gi.c"], COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_gi.c"],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论