提交 509f94a3 authored 作者: Frederic's avatar Frederic

Fix pickling of GpuConv op due to not completly removing set_flops last time.

上级 601088bb
...@@ -252,12 +252,6 @@ AddConfigVar('lib.amdlibm', ...@@ -252,12 +252,6 @@ AddConfigVar('lib.amdlibm',
"Use amd's amdlibm numerical library", "Use amd's amdlibm numerical library",
BoolParam(False)) BoolParam(False))
AddConfigVar('op.set_flops',
("currently used only in ConvOp. The profile mode will print the "
"flops/s for the op."),
BoolParam(False),
in_c_key=False)
AddConfigVar('gpuelemwise.sync', AddConfigVar('gpuelemwise.sync',
"when true, wait that the gpu fct finished and check it error code.", "when true, wait that the gpu fct finished and check it error code.",
BoolParam(True), BoolParam(True),
......
...@@ -1029,9 +1029,6 @@ def local_gpu_conv(node): ...@@ -1029,9 +1029,6 @@ def local_gpu_conv(node):
verbose=op.verbose, verbose=op.verbose,
imshp=op.imshp, imshp=op.imshp,
) )
#HACK to print the number of MFlops in the profiler output.
if hasattr(op, 'flops'):
ret.flops = op.flops
if op.imshp_logical is not None: if op.imshp_logical is not None:
logical_img_hw = op.imshp_logical[1:3] logical_img_hw = op.imshp_logical[1:3]
if logical_img_hw != op.imshp[1:3]: if logical_img_hw != op.imshp[1:3]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论