Skip to content

P
pytensor
提交 2a54b6c3 authored 作者: Frederic's avatar Frederic
浏览文件
操作

Fix Elemwise and CAReduce version when the scalar_op don't have a version.

We were not returning () when the scalar op don't have a version.
上级 29968300
隐藏空白字符变更
内嵌 并排
正在显示 包含 4 行增加 和 4 行删除
theano/tensor/elemwise.py
浏览文件 @ 2a54b6c3
...@@ -1146,9 +1146,9 @@ class Elemwise(Op): ...@@ -1146,9 +1146,9 @@ class Elemwise(Op):
scalar_node = Apply(self.scalar_op, scalar_node = Apply(self.scalar_op,
[Scalar(dtype=input.type.dtype)() for input in node.inputs], [Scalar(dtype=input.type.dtype)() for input in node.inputs],
[Scalar(dtype=output.type.dtype)() for output in node.outputs]) [Scalar(dtype=output.type.dtype)() for output in node.outputs])
version.extend(self.scalar_op.c_code_cache_version_apply(scalar_node)) version.append(self.scalar_op.c_code_cache_version_apply(scalar_node))
for i in node.inputs + node.outputs: for i in node.inputs + node.outputs:
version.extend(Scalar(dtype=i.type.dtype).c_code_cache_version()) version.append(Scalar(dtype=i.type.dtype).c_code_cache_version())
if all(version): if all(version):
return tuple(version) return tuple(version)
else: else:
...@@ -1561,9 +1561,9 @@ for(int i=0;i<PyArray_NDIM(%(iname)s);i++){ ...@@ -1561,9 +1561,9 @@ for(int i=0;i<PyArray_NDIM(%(iname)s);i++){
scalar_node = Apply(self.scalar_op, scalar_node = Apply(self.scalar_op,
[Scalar(dtype=input.type.dtype)() for input in node.inputs], [Scalar(dtype=input.type.dtype)() for input in node.inputs],
[Scalar(dtype=output.type.dtype)() for output in node.outputs]) [Scalar(dtype=output.type.dtype)() for output in node.outputs])
version.extend(self.scalar_op.c_code_cache_version_apply(scalar_node)) version.append(self.scalar_op.c_code_cache_version_apply(scalar_node))
for i in node.inputs + node.outputs: for i in node.inputs + node.outputs:
version.extend(Scalar(dtype=i.type.dtype).c_code_cache_version()) version.append(Scalar(dtype=i.type.dtype).c_code_cache_version())
if all(version): if all(version):
return tuple(version) return tuple(version)
else: else:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论