提交 2b0b7acd authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Fix c code and flake8 formatting

上级 81f0f196
...@@ -1740,6 +1740,7 @@ def local_gpu_max_pool_rop(op, ctx_name, inputs, outputs): ...@@ -1740,6 +1740,7 @@ def local_gpu_max_pool_rop(op, ctx_name, inputs, outputs):
ret_padded = op(inp_padded, eval_inp_padded, ws, stride, pad) ret_padded = op(inp_padded, eval_inp_padded, ws, stride, pad)
return unpad_dims(ret_padded, inp, 2, nd) return unpad_dims(ret_padded, inp, 2, nd)
@register_opt("low_memory") @register_opt("low_memory")
@local_optimizer([GpuCAReduceCuda]) @local_optimizer([GpuCAReduceCuda])
def local_gpu_elemwise_careduce(node): def local_gpu_elemwise_careduce(node):
......
...@@ -285,16 +285,18 @@ class test_RopLop(RopLop_checker): ...@@ -285,16 +285,18 @@ class test_RopLop(RopLop_checker):
maxpool_op = Pool(ignore_border, ndim=len(ws)) maxpool_op = Pool(ignore_border, ndim=len(ws))
a_pooled = maxpool_op(x, ws).flatten() a_pooled = maxpool_op(x, ws).flatten()
yv = tensor.Rop(a_pooled, x, ex) yv = tensor.Rop(a_pooled, x, ex)
rop_f = function([], yv, on_unused_input='ignore') mode = None
if theano.config.mode == "FAST_COMPILE":
mode = "FAST_RUN"
rop_f = function([], yv, on_unused_input='ignore', mode=mode)
sy, _ = theano.scan(lambda i, y, x, v: sy, _ = theano.scan(lambda i, y, x, v:
(tensor.grad(y[i], x) * v).sum(), (tensor.grad(y[i], x) * v).sum(),
sequences=tensor.arange(a_pooled.shape[0]), sequences=tensor.arange(a_pooled.shape[0]),
non_sequences=[a_pooled, x, ex]) non_sequences=[a_pooled, x, ex])
scan_f = function([], sy, on_unused_input='ignore') scan_f = function([], sy, on_unused_input='ignore', mode=mode)
v1 = rop_f() v1 = rop_f()
v2 = scan_f() v2 = scan_f()
assert numpy.allclose(v1, v2), ("Rop mismatch: %s %s" % assert numpy.allclose(v1, v2), ("Rop mismatch: %s %s" % (v1, v2))
(v1, v2))
def test_conv(self): def test_conv(self):
for conv_op in [conv.conv2d, conv2d]: for conv_op in [conv.conv2d, conv2d]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论