One case for dyadic constant-each

This commit is contained in:
Marshall Lochbaum 2023-05-18 15:40:52 -04:00
parent ed5fac7cc9
commit 93e590fd9d

View File

@ -38,6 +38,10 @@ B eachd_fn(B fo, B w, B x, FC2 f) {
B rb; B rb;
if (ria==0) { if (ria==0) {
rb = rM==1? emptyHVec() : m_harrUc(bo).b; rb = rM==1? emptyHVec() : m_harrUc(bo).b;
} else if (f == arr_c2) {
MAKE_MUT(rm, ria);
mut_fill(rm, 0, fo, ria);
rb = mut_fc(rm, bo);
} else { } else {
M_HARR(r, ria) M_HARR(r, ria)
if (wr==xr) { SGet(x) SGet(w) for(usz ri=0; ri<ria; ri++) HARR_ADD(r, ri, f(fo, Get(w,ri), Get(x,ri))); } if (wr==xr) { SGet(x) SGet(w) for(usz ri=0; ri<ria; ri++) HARR_ADD(r, ri, f(fo, Get(w,ri), Get(x,ri))); }