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

fix a bug temporally while I found the root of the cause.

上级 02df7759
......@@ -670,7 +670,7 @@ class NaiveAlgo(object):
for(int i=nd-1;i>0;i--){
if(dims[i]==1 && strides[i]==0){//
collapse[i]=1;
}else if(dims[i-1]==1 && strides[i-1]==0){
}else if(false && dims[i-1]==1 && strides[i-1]==0){
collapse[i]=1;
}else if(strides[i]*dims[i]==strides[i-1]){//the dims nd-1 are not strided again dimension nd
collapse[i]=1;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论