提交 667d2398 authored 作者: Frederic's avatar Frederic

Fuse less elemwise when no compiler is available. (fix test crash in that case)

上级 585bf1b9
......@@ -5057,7 +5057,17 @@ your code will run correctly, but may be slower.""")
return n.outputs
return local_fuse
local_elemwise_fusion = local_elemwise_fusion_op(T.Elemwise)
def elemwise_max_input_fct(node):
# The Elemwise.perform use numpy ufunc and they are limited to 32
# inputs.
if not theano.config.cxx:
return 32
return 1024
local_elemwise_fusion = local_elemwise_fusion_op(T.Elemwise,
elemwise_max_input_fct)
class FusionOptimizer(Optimizer):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论