提交 e1fda94f authored 作者: Frederic Bastien's avatar Frederic Bastien

fix gpuelemwise when their is 0 dimensions. Why this case happen?

上级 51c892be
...@@ -848,9 +848,11 @@ nd_collapse_[i]=0; ...@@ -848,9 +848,11 @@ nd_collapse_[i]=0;
for(int i=1, end=nd_collapse;i<end;i++){ for(int i=1, end=nd_collapse;i<end;i++){
if(nd_collapse_[i]==1)nd_collapse--; if(nd_collapse_[i]==1)nd_collapse--;
} }
if(nd_collapse == 1 && """%locals() if(nd_collapse == 1 """%locals()
print >> sio, " && ".join(["local_str[%(ipos)s][nd_collapse-1]==1 "%locals()for ipos in range(len(node.inputs)) if not _logical_scalar(node.inputs[ipos])]+ l=["local_str[%(ipos)s][nd_collapse-1]==1 "%locals()for ipos in range(len(node.inputs)) if not _logical_scalar(node.inputs[ipos])]
["local_ostr[%(ipos)s][nd_collapse-1]==1 "%locals()for ipos in range(len(node.outputs)) if not _logical_scalar(node.outputs[ipos])]) l+=["local_ostr[%(ipos)s][nd_collapse-1]==1 "%locals()for ipos in range(len(node.outputs)) if not _logical_scalar(node.outputs[ipos])]
if len(l)>0:
print >> sio," && "," && ".join(l)
print >> sio,"""){nd_collapse=0;} """ print >> sio,"""){nd_collapse=0;} """
if self.verbose: if self.verbose:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论