add missing shape equality check to fast dyadic arith

This commit is contained in:
dzaima 2021-05-28 13:48:59 +03:00
parent 23d823057a
commit 9326aa9572

View File

@ -24,6 +24,7 @@
EXTRA \
if (isArr(w)|isArr(x)) { B ow=w; B ox=x; \
if (isArr(w)&isArr(x) && rnk(w)==rnk(x)) { \
if (memcmp(a(w)->sh, a(x)->sh, rnk(w)*sizeof(usz))) thrF(#NAME ": Expected equal shape prefix (%H ≡ ≢𝕨, %H ≡ ≢𝕩)", w, x); \
usz ia = a(x)->ia; \
u8 we = TI(w).elType; \
u8 xe = TI(x).elType; \
@ -94,6 +95,7 @@
EXTRA \
if (isArr(w)|isArr(x)) { \
if (isArr(w)&isArr(x) && rnk(w)==rnk(x)) { \
if (memcmp(a(w)->sh, a(x)->sh, rnk(w)*sizeof(usz))) thrF(#NAME ": Expected equal shape prefix (%H ≡ ≢𝕨, %H ≡ ≢𝕩)", w, x); \
usz ia = a(x)->ia; \
u8 we = TI(w).elType; \
u8 xe = TI(x).elType; \