提交 740295e2 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make the core code for the convolution reusable.

上级 c022347b
差异被折叠。
...@@ -4,7 +4,7 @@ Tests for GPU convolution ...@@ -4,7 +4,7 @@ Tests for GPU convolution
import sys import sys
import time import time
import unittest import unittest
import traceback
import numpy import numpy
...@@ -286,7 +286,7 @@ def exec_conv(version, shapes, verbose, random, mode, ...@@ -286,7 +286,7 @@ def exec_conv(version, shapes, verbose, random, mode,
cls=cls) cls=cls)
except Exception, e: except Exception, e:
print ver, id, (ishape, kshape, subshape, istride, kstride) print ver, id, (ishape, kshape, subshape, istride, kstride)
print "Exception", type(e), e print traceback.format_exc()
pass pass
if not ret: if not ret:
failed_version.add(ver) failed_version.add(ver)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论