diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2025-06-21 09:57:45 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2025-06-21 09:57:45 +0200 |
| commit | b5ed3d2fcc249cb410b9e86d25e9ef808c6dba97 (patch) | |
| tree | 66383b16d5d95f939aaa165a783dbbfd99a57fe3 /src/Language.hs | |
| parent | 8bbc2d2867e3d0a4a1f2810b40e92175779822e1 (diff) | |
| parent | a4b3eb76acbec30ffeae119a4dc6e4c9f64396fe (diff) | |
Diffstat (limited to 'src/Language.hs')
| -rw-r--r-- | src/Language.hs | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/Language.hs b/src/Language.hs index 7a780a0..4e6d604 100644 --- a/src/Language.hs +++ b/src/Language.hs @@ -17,6 +17,7 @@ module Language ( import Array import AST +import AST.Sparse.Types import AST.Types import CHAD.Types import Data @@ -175,8 +176,11 @@ recompute = NERecompute with :: forall t a env acname. KnownMTy t => NExpr env t -> (Var acname (TAccum t) :-> NExpr ('(acname, TAccum t) : env) a) -> NExpr env (TPair a t) with a (n :-> b) = NEWith (knownMTy @t) a n b -accum :: KnownMTy t => SAcPrj p t a -> NExpr env (AcIdx p t) -> NExpr env a -> NExpr env (TAccum t) -> NExpr env TNil -accum p a b c = NEAccum knownMTy p a b c +accum :: KnownMTy t => SAcPrj p t a -> NExpr env (AcIdxD p t) -> NExpr env a -> NExpr env (TAccum t) -> NExpr env TNil +accum p a b c = NEAccum knownMTy p a (spDense (acPrjTy p knownMTy)) b c + +accumS :: KnownMTy t => SAcPrj p t a -> NExpr env (AcIdxD p t) -> Sparse a b -> NExpr env b -> NExpr env (TAccum t) -> NExpr env TNil +accumS p a sp b c = NEAccum knownMTy p a sp b c (.==) :: (KnownScalTy st, ScalIsNumeric st ~ True) => NExpr env (TScal st) -> NExpr env (TScal st) -> NExpr env (TScal TBool) |