Test requiring outer split + unforkify

juno_samples/fork_join_tests/src/fork_join_tests.jn

@@ -44,3 +44,18 @@ fn test3(input : i32) -> i32[3, 3] {
   }
   return arr3;
 }
+
+#[entry]
+fn test4(input : i32) -> i32[4, 4] {
+  let arr : i32[4, 4];
+  for i = 0 to 4 {
+    for j = 0 to 4 {
+      let acc = arr[i, j];
+      for k = 0 to 7 {
+        acc += input;
+      }
+      arr[i, j] = acc;
+    }
+  }
+  return arr;
+}

juno_samples/fork_join_tests/src/gpu.sch

@@ -6,6 +6,7 @@ let out = auto-outline(*);
 gpu(out.test1);
 gpu(out.test2);
 gpu(out.test3);
+gpu(out.test4);
 
 ip-sroa(*);
 sroa(*);

juno_samples/fork_join_tests/src/main.rs

@@ -32,6 +32,11 @@ fn main() {
         let output = r.run(0).await;
         let correct = vec![11, 10, 9, 10, 9, 8, 9, 8, 7];
         assert(correct, output);
+
+        let mut r = runner!(test4);
+        let output = r.run(9).await;
+        let correct = vec![63i32; 16];
+        assert(correct, output);
     });
 }