From 3dcb32a54a9238be92f01c20d4f4ad17f37bc0a1 Mon Sep 17 00:00:00 2001 From: dzaima Date: Sat, 6 May 2023 16:27:10 +0300 Subject: [PATCH] =?UTF-8?q?fast=20Arith=E2=8E=89+n=20where=20n=20is=20the?= =?UTF-8?q?=20rank=20of=20one=20argument?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/builtins/cells.c | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/src/builtins/cells.c b/src/builtins/cells.c index dac678ee..9d22b01d 100644 --- a/src/builtins/cells.c +++ b/src/builtins/cells.c @@ -4,7 +4,7 @@ #include "../utils/calls.h" #include -B shape_c1(B, B); +B fne_c1(B, B); B shape_c2(B, B, B); B transp_c2(B, B, B); B fold_rows(Md1D* d, B x); // from fold.c @@ -480,15 +480,26 @@ static NOINLINE B rank2_empty(B f, B w, ur wk, B x, ur xk, u32 chr) { } NOINLINE B for_cells_AS(B f, B w, B x, ur wcr, ur wr, u32 chr) { + assert(isArr(w)); ur wk = wr-wcr; assert(wk>0 && wcr0 && xcr