Some more primitive operators

1 files changed, 6 insertions, 0 deletions

diff --git a/src/AST.hs b/src/AST.hs
index 28c5b37..e3da634 100644
--- a/src/AST.hs
+++ b/src/AST.hs
@@ -169,6 +169,9 @@ data SOp a t where
   OIf :: SOp (TScal TBool) (TEither TNil TNil)  -- True is Left, False is Right
   ORound64 :: SOp (TScal TF64) (TScal TI64)
   OToFl64 :: SOp (TScal TI64) (TScal TF64)
+  ORecip :: ScalIsFloating a ~ True => SScalTy a -> SOp (TScal a) (TScal a)
+  OExp :: ScalIsFloating a ~ True => SScalTy a -> SOp (TScal a) (TScal a)
+  OLog :: ScalIsFloating a ~ True => SScalTy a -> SOp (TScal a) (TScal a)
 deriving instance Show (SOp a t)
 
 opt2 :: SOp a t -> STy t
@@ -185,6 +188,9 @@ opt2 = \case
   OIf -> STEither STNil STNil
   ORound64 -> STScal STI64
   OToFl64 -> STScal STF64
+  ORecip t -> STScal t
+  OExp t -> STScal t
+  OLog t -> STScal t
 
 typeOf :: Expr x env t -> STy t
 typeOf = \case