fix inferlocations

gibbon-compiler/src/Gibbon/Passes/InferLocations.hs

@@ -1427,10 +1427,11 @@ fixProj renam pvar proj e =
 
 
 -- Runs after projTups in the SpawnE case in inferExp.
-moveProjsAfterSync :: Var -> Exp2 -> Exp2
-moveProjsAfterSync sv ex = go [] (S.singleton sv) ex
+moveProjsAfterSync :: LocVar -> Exp2 -> Exp2
+moveProjsAfterSync sv ex = case sv of 
+                                l@(Single loc) -> go [] (S.singleton l) ex
   where
-    go :: [Binds (Exp2)] -> S.Set Var -> Exp2 -> Exp2
+    go :: [Binds (Exp2)] -> S.Set LocVar -> Exp2 -> Exp2
     go acc1 pending ex =
       case ex of
         VarE{}    -> ex
@@ -1447,7 +1448,7 @@ moveProjsAfterSync sv ex = go [] (S.singleton sv) ex
           let vars = allFreeVars rhs
           in if S.null (S.intersection vars pending)
              then LetE (v, locs, ty, rhs) (go acc1 pending bod)
-             else go ((v, locs, ty, rhs):acc1) (S.insert v pending) bod
+             else go ((v, locs, ty, rhs):acc1) (S.insert (singleLocVar v) pending) bod
         IfE a b c   -> IfE (go acc1 pending a) (go acc1 pending b) (go acc1 pending c)
         MkProdE ls  -> MkProdE $ L.map (go acc1 pending) ls
         ProjE i arg -> ProjE i $ go acc1 pending arg