add threshold for reshape-based ⎉+n path
This commit is contained in:
parent
3dcb32a54a
commit
d62c77af41
@ -488,7 +488,7 @@ NOINLINE B for_cells_AS(B f, B w, B x, ur wcr, ur wr, u32 chr) {
|
||||
if (isPervasiveDy(f)) {
|
||||
if (isAtm(x)) return c2(f, w, x);
|
||||
if (RNK(x)!=wcr || !eqShPart(SH(x), wsh+wk, wcr)) goto generic;
|
||||
if (TI(w,elType)==el_B || TI(x,elType)==el_B) goto generic;
|
||||
if (TI(w,elType)==el_B || TI(x,elType)==el_B || (IA(x)>500 && IA(w)!=IA(x))) goto generic;
|
||||
return c2(f, w, C2(shape, C1(fne, incG(w)), x));
|
||||
}
|
||||
}
|
||||
@ -525,7 +525,7 @@ NOINLINE B for_cells_SA(B f, B w, B x, ur xcr, ur xr, u32 chr) {
|
||||
if (isPervasiveDy(f)) {
|
||||
if (isAtm(w)) return c2(f, w, x);
|
||||
if (RNK(w)!=xcr || !eqShPart(SH(w), xsh+xk, xcr)) goto generic;
|
||||
if (TI(w,elType)==el_B || TI(x,elType)==el_B) goto generic;
|
||||
if (TI(w,elType)==el_B || TI(x,elType)==el_B || (IA(w)>500 && IA(w)!=IA(x))) goto generic;
|
||||
return c2(f, C2(shape, C1(fne, incG(x)), w), x);
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user