summaryrefslogtreecommitdiff
path: root/src/Simplify.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Simplify.hs')
-rw-r--r--src/Simplify.hs2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/Simplify.hs b/src/Simplify.hs
index cfbdbb9..0ce5594 100644
--- a/src/Simplify.hs
+++ b/src/Simplify.hs
@@ -111,6 +111,7 @@ simplify' = \case
EAccum i e1 e2 e3 -> EAccum i <$> simplify' e1 <*> simplify' e2 <*> simplify' e3
EZero t -> pure $ EZero t
EPlus t a b -> EPlus t <$> simplify' a <*> simplify' b
+ EOneHot t i a b -> EOneHot t i <$> simplify' a <*> simplify' b
EError t s -> pure $ EError t s
acted :: (Any, a) -> (Any, a)
@@ -156,6 +157,7 @@ hasAdds = \case
EAccum _ _ _ _ -> True
EZero _ -> False
EPlus _ a b -> hasAdds a || hasAdds b
+ EOneHot _ _ a b -> hasAdds a || hasAdds b
EError _ _ -> False
checkAccumInScope :: SList STy env -> Bool