提交 9840faaf authored 作者: Frederic Bastien's avatar Frederic Bastien

As we also do indexing in int64, lower the fusion even more

上级 79a344c5
...@@ -47,7 +47,8 @@ def max_inputs_to_GpuElemwise(node_or_outputs): ...@@ -47,7 +47,8 @@ def max_inputs_to_GpuElemwise(node_or_outputs):
else: else:
outputs = node_or_outputs outputs = node_or_outputs
ptr_size = 8 ptr_size = 8
int_size = 4 # We compile code that would do indexing in int64
int_size = 8
# we take the limit from CUDA for now # we take the limit from CUDA for now
argument_limit = 232 argument_limit = 232
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论