Complete GMM implementation

1 files changed, 2 insertions, 0 deletions

diff --git a/src/AST.hs b/src/AST.hs
index 263b806..e7dde90 100644
--- a/src/AST.hs
+++ b/src/AST.hs
@@ -172,6 +172,7 @@ data SOp a t where
   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)
+  OIDiv :: ScalIsIntegral a ~ True => SScalTy a -> SOp (TPair (TScal a) (TScal a)) (TScal a)
 deriving instance Show (SOp a t)
 
 opt2 :: SOp a t -> STy t
@@ -191,6 +192,7 @@ opt2 = \case
   ORecip t -> STScal t
   OExp t -> STScal t
   OLog t -> STScal t
+  OIDiv t -> STScal t
 
 typeOf :: Expr x env t -> STy t
 typeOf = \case