提交 537c9bdd authored 作者: Razvan Pascanu's avatar Razvan Pascanu

correction to the cython version of code

上级 ee89ad32
...@@ -528,6 +528,10 @@ class Scan(PureOp): ...@@ -528,6 +528,10 @@ class Scan(PureOp):
dtype='int32') dtype='int32')
cython_vector_outs = numpy.asarray(self.vector_outs, cython_vector_outs = numpy.asarray(self.vector_outs,
dtype='int32') dtype='int32')
cython_destroy_map = [x in self.destroy_map
for x in xrange(len(node.outputs))]
cython_destroy_map = numpy.asarray(cython_destroy_map,
dtype='int32')
import scan_perform_ext import scan_perform_ext
p = lambda node, args, outs:\ p = lambda node, args, outs:\
scan_perform_ext.perform( scan_perform_ext.perform(
...@@ -549,6 +553,7 @@ class Scan(PureOp): ...@@ -549,6 +553,7 @@ class Scan(PureOp):
cython_mit_mot_out_nslices, cython_mit_mot_out_nslices,
self.fn.fn, self.fn.fn,
self.fn, self.fn,
cython_destroy_map,
args, args,
outs, outs,
self) self)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论