Consider sel cheap for inlining in Expand.

Feldspar · Mar 17, 2021 · a55d3c0 · a55d3c0
1 parent 6f03cdd
commit a55d3c0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Feldspar/Core/Middleend/Expand.hs b/src/Feldspar/Core/Middleend/Expand.hs
@@ -237,8 +237,11 @@ ixAndInv ai v vs = go ai
 
 -- | Check that an expression is a simple array reference that we can inline
 simpleArrRef :: UExp -> Bool
-simpleArrRef (In _ (App GetIx _ [In _ (Variable _), e])) = simpleIdxE e
+simpleArrRef (In _ (App GetIx _ [a, e])) = simpleArrE a && simpleIdxE e
   where simpleIdxE (In _ (Variable _)) = True
         simpleIdxE (In _ (App op _ es)) = op `elem` [Add,Sub,Mul] && all simpleIdxE es
         simpleIdxE _ = False
+        simpleArrE (In _ (Variable _)) = True
+        simpleArrE (In _ (App (Sel _) _ [In _ (Variable _)])) = True
+        simpleArrE _ = False
 simpleArrRef _ = False