matching el_i8 and el_i16 cmp impls

This commit is contained in:
dzaima 2021-09-05 22:33:36 +03:00
parent 4e883f4812
commit 45c2fb41cc
2 changed files with 15 additions and 1 deletions

View File

@ -15,6 +15,20 @@
if (wa&xa && !eqShape(w, x)) thrF(CHR": Expected equal shape prefix (%H ≡ ≢𝕨, %H ≡ ≢𝕩)", w, x); \
u8 we = wa? TI(w,elType) : selfElType(w); \
u8 xe = xa? TI(x,elType) : selfElType(x); \
if (we==el_i8 && xe==el_i8) { \
i8* rp; B r = m_i8arrc(&rp, isArr(x)? x : w); usz ria=a(r)->ia; /* (i8)o2iu safe bc _e==el_i8 */ \
if (!wa) { i8 wv=o2iu(w); i8* xp=i8any_ptr(x); for(usz i=0;i<ria;i++)rp[i]=wv OP xp[i]; } \
else if (!xa) { i8 xv=o2iu(x); i8* wp=i8any_ptr(w); for(usz i=0;i<ria;i++)rp[i]=wp[i] OP xv; } \
else { i8* wp=i8any_ptr(w); i8* xp=i8any_ptr(x); for(usz i=0;i<ria;i++)rp[i]=wp[i] OP xp[i]; } \
if(wa) dec(w); if(xa) dec(x); return r; \
} \
if (we==el_i16 && xe==el_i16) { \
i8* rp; B r = m_i8arrc(&rp, isArr(x)? x : w); usz ria=a(r)->ia; \
if (!wa) { i16 wv=o2iu(w); i16* xp=i16any_ptr(x); for(usz i=0;i<ria;i++)rp[i]=wv OP xp[i]; } \
else if (!xa) { i16 xv=o2iu(x); i16* wp=i16any_ptr(w); for(usz i=0;i<ria;i++)rp[i]=wp[i] OP xv; } \
else { i16* wp=i16any_ptr(w); i16* xp=i16any_ptr(x); for(usz i=0;i<ria;i++)rp[i]=wp[i] OP xp[i]; } \
if(wa) dec(w); if(xa) dec(x); return r; \
} \
if (we==el_i32 && xe==el_i32) { \
i8* rp; B r = m_i8arrc(&rp, isArr(x)? x : w); usz ria=a(r)->ia; \
if (!wa) { i32 wv=o2iu(w); i32* xp=i32any_ptr(x); for(usz i=0;i<ria;i++)rp[i]=wv OP xp[i]; } \

View File

@ -547,7 +547,7 @@ static u8 fillElType(B x) {
return el_B;
}
static u8 selfElType(B x) { // guaranteed to fit fill
if (isF64(x)) return q_i32(x)? el_i32 : el_f64;
if (isF64(x)) return q_i16(x)? (q_i8(x)? el_i8 : el_i16) : (q_i32(x)? el_i32 : el_f64);
if (isC32(x)) return el_c32;
return el_B;
}