gistfile1.hs

sfold :: (Shape ix, Elt a) => (Exp a -> Exp a -> Exp a) -> Exp a -> Acc (Array ix a) -> Exp a
sfold f i a = let
                    iteratefunc :: (Elt a) => (Exp a -> Exp a -> Exp a) -> Exp a -> Acc (Vector a) -> Exp a
                    iteratefunc !f !i !ar =
                                    ((size ar) ==* 0) ? (i,
                                      let
                                        hd = A.take (constant 1) ar A.!! 0
                                        tl = A.tail ar
                                      in
                                      f hd (iteratefunc f i tl))
              in
                    iteratefunc f i (flatten a)