diff --git a/src/Compiler/Transform.idr b/src/Compiler/Transform.idr index cfec11976..f3ef66723 100644 --- a/src/Compiler/Transform.idr +++ b/src/Compiler/Transform.idr @@ -89,8 +89,8 @@ call (MkFn params (Cond pred ft t ff f)) args = ?call_cond call (MkFn params (Dot x y)) args = Dot (call (MkFn params x) args) (call (MkFn params y) args) call (MkFn params (Cholesky x)) args = Cholesky (call (MkFn params x) args) call (MkFn params (TriangularSolve x y lower)) args = TriangularSolve (call (MkFn params x) args) (call (MkFn params y) args) lower -call (MkFn params (UniformFloatingPoint x y z w xs)) args = ?call_uniformFloatingPoint -call (MkFn params (NormalFloatingPoint x y xs)) args = ?call_normalFloatingPoint +call (MkFn params (UniformFloatingPoint key state minval maxval shape)) args = UniformFloatingPoint (call (MkFn params key) args) (call (MkFn params state) args) (call (MkFn params minval) args) (call (MkFn params maxval) args) shape +call (MkFn params (NormalFloatingPoint key state shape)) args = NormalFloatingPoint (call (MkFn params key) args) (call (MkFn params state) args) shape -- what if there are parameters in the expression from the surrounding scope? like if we use `vmap` in `sort`? --