kumi 0.0.18 → 0.0.20

data/golden/reduction_broadcast/expected/lir_06_const_prop.txt

@@ -0,0 +1,68 @@
+(LIR
+  (Declaration dept_headcount
+    %t1 = load_input "departments" :: array
+    loop departments id=L1 in %t1 as el=%departments_el_2, idx=%departments_i_3
+      %acc_4 = decl_acc :: integer
+      %t5 = load_field departments_el_2["teams"] :: any
+      loop teams id=L2 in %t5 as el=%teams_el_6, idx=%teams_i_7
+        %t8 = load_field teams_el_6["headcount"] :: integer
+        %acc_4 = acc_add agg.sum(%acc_4, %t8) :: integer
+      end_loop
+      %t9 = acc_load %acc_4 :: integer
+      yield %t9
+    end_loop
+  )
+  (Declaration teams_per_dept
+    %t10 = load_input "departments" :: array
+    loop departments id=L3 in %t10 as el=%departments_el_11, idx=%departments_i_12
+      %acc_13 = decl_acc :: integer
+      %t14 = load_field departments_el_11["teams"] :: any
+      loop teams id=L4 in %t14 as el=%teams_el_15, idx=%teams_i_16
+        %t17 = load_field teams_el_15["team_name"] :: string
+        %acc_13 = acc_add agg.count(%acc_13, %t17) :: integer
+      end_loop
+      %t18 = acc_load %acc_13 :: integer
+      yield %t18
+    end_loop
+  )
+  (Declaration avg_headcount_per_dept
+    %t19 = load_input "departments" :: array
+    loop departments id=L5 in %t19 as el=%departments_el_20, idx=%departments_i_21
+      %acc36 = decl_acc :: integer
+      %t37 = load_field departments_el_20["teams"] :: any
+      %acc42 = decl_acc :: integer
+      loop teams id=L8 in %t37 as el=%teams_el_6, idx=%teams_i_7
+        %t38 = load_field teams_el_6["headcount"] :: integer
+        %acc36 = acc_add agg.sum(%acc36, %t38) :: integer
+        %t44 = load_field teams_el_6["team_name"] :: string
+        %acc42 = acc_add agg.count(%acc42, %t44) :: integer
+      end_loop
+      %t39 = acc_load %acc36 :: integer
+      %t45 = acc_load %acc42 :: integer
+      %t24 = call core.div(%t39, %t45) :: float
+      yield %t24
+    end_loop
+  )
+  (Declaration is_above_average_team
+    %t25 = load_input "departments" :: array
+    loop departments id=L6 in %t25 as el=%departments_el_26, idx=%departments_i_27
+      %t28 = load_field departments_el_26["teams"] :: any
+      %acc52 = decl_acc :: integer
+      %acc58 = decl_acc :: integer
+      loop teams id=L10 in %t28 as el=%teams_el_6, idx=%teams_i_7
+        %t54 = load_field teams_el_6["headcount"] :: integer
+        %acc52 = acc_add agg.sum(%acc52, %t54) :: integer
+        %t60 = load_field teams_el_6["team_name"] :: string
+        %acc58 = acc_add agg.count(%acc58, %t60) :: integer
+      end_loop
+      %t55 = acc_load %acc52 :: integer
+      %t61 = acc_load %acc58 :: integer
+      %t49 = call core.div(%t55, %t61) :: float
+      loop teams id=L7 in %t28 as el=%teams_el_29, idx=%teams_i_30
+        %t31 = load_field teams_el_29["headcount"] :: integer
+        %t33 = call core.gt(%t31, %t49) :: boolean
+        yield %t33
+      end_loop
+    end_loop
+  )
+)

data/golden/reduction_broadcast/expected/nast.txt

@@ -0,0 +1,24 @@
+(NAST
+  (VALUE dept_headcount
+    (Call :"agg.sum"
+    (InputRef [:departments, :dept, :teams, :team, :headcount])
+  )
+  )
+  (VALUE teams_per_dept
+    (Call :"agg.count"
+    (InputRef [:departments, :dept, :teams, :team, :team_name])
+  )
+  )
+  (VALUE avg_headcount_per_dept
+    (Call :"core.div"
+    (Ref dept_headcount)
+    (Ref teams_per_dept)
+  )
+  )
+  (TRAIT is_above_average_team
+    (Call :"core.gt"
+    (InputRef [:departments, :dept, :teams, :team, :headcount])
+    (Ref avg_headcount_per_dept)
+  )
+  )
+)

data/golden/reduction_broadcast/expected/schema_javascript.mjs

@@ -0,0 +1,74 @@
+export function _dept_headcount(input) {
+  let out = [];
+  let t1 = input["departments"];
+  t1.forEach((departments_el_2, departments_i_3) => {
+    let acc_4 = 0;
+    let t5 = departments_el_2["teams"];
+    t5.forEach((teams_el_6, teams_i_7) => {
+      let t8 = teams_el_6["headcount"];
+      acc_4 += t8;
+    });
+    out.push(acc_4);
+  });
+  return out;
+}
+
+export function _teams_per_dept(input) {
+  let out = [];
+  let t10 = input["departments"];
+  t10.forEach((departments_el_11, departments_i_12) => {
+    let acc_13 = 0;
+    let t14 = departments_el_11["teams"];
+    t14.forEach((teams_el_15, teams_i_16) => {
+      let t17 = teams_el_15["team_name"];
+      acc_13 += 1;
+    });
+    out.push(acc_13);
+  });
+  return out;
+}
+
+export function _avg_headcount_per_dept(input) {
+  let out = [];
+  let t19 = input["departments"];
+  t19.forEach((departments_el_20, departments_i_21) => {
+    let acc36 = 0;
+    let t37 = departments_el_20["teams"];
+    let acc42 = 0;
+    t37.forEach((teams_el_6, teams_i_7) => {
+      let t38 = teams_el_6["headcount"];
+      acc36 += t38;
+      let t44 = teams_el_6["team_name"];
+      acc42 += 1;
+    });
+    let t24 = acc36 / acc42;
+    out.push(t24);
+  });
+  return out;
+}
+
+export function _is_above_average_team(input) {
+  let out = [];
+  let t25 = input["departments"];
+  t25.forEach((departments_el_26, departments_i_27) => {
+    let out_1 = [];
+    let t28 = departments_el_26["teams"];
+    let acc52 = 0;
+    let acc58 = 0;
+    t28.forEach((teams_el_6, teams_i_7) => {
+      let t54 = teams_el_6["headcount"];
+      acc52 += t54;
+      let t60 = teams_el_6["team_name"];
+      acc58 += 1;
+    });
+    let t49 = acc52 / acc58;
+    t28.forEach((teams_el_29, teams_i_30) => {
+      let t31 = teams_el_29["headcount"];
+      let t33 = t31 > t49;
+      out_1.push(t33);
+    });
+    out.push(out_1);
+  });
+  return out;
+}
+

data/golden/reduction_broadcast/expected/schema_ruby.rb

@@ -0,0 +1,110 @@
+# Autogenerated by Kumi Codegen
+module Kumi::Compiled::KUMI_621112a9f96fd9df9274375684d4aad116dd930db67521f72ec03d1f911fd1ca
+  def self.from(input_data = nil)
+    instance = Object.new
+    instance.extend(self)
+    instance.instance_variable_set(:@input, input_data)
+    instance
+  end
+
+  def self.__kumi_executable__
+    instance = Object.new
+    instance.extend(self)
+    instance
+  end
+
+  def update(input_data)
+    @input = @input.merge(input_data)
+    self
+  end
+
+  def [](name)
+    case name
+    when :dept_headcount then _dept_headcount
+    when :teams_per_dept then _teams_per_dept
+    when :avg_headcount_per_dept then _avg_headcount_per_dept
+    when :is_above_average_team then _is_above_average_team
+    else raise KeyError, "Unknown declaration"
+    end
+  end
+
+  def _dept_headcount(input = @input)
+    out = []
+    t1 = input["departments"] || input[:departments]
+    t1.each_with_index do |departments_el_2, departments_i_3|
+      acc_4 = 0
+      t5 = departments_el_2["teams"] || departments_el_2[:teams]
+      t5.each_with_index do |teams_el_6, teams_i_7|
+        t8 = teams_el_6["headcount"] || teams_el_6[:headcount]
+        acc_4 += t8
+      end
+      t9 = acc_4
+      out << t9
+    end
+    out
+  end
+
+  def _teams_per_dept(input = @input)
+    out = []
+    t10 = input["departments"] || input[:departments]
+    t10.each_with_index do |departments_el_11, departments_i_12|
+      acc_13 = 0
+      t14 = departments_el_11["teams"] || departments_el_11[:teams]
+      t14.each_with_index do |teams_el_15, teams_i_16|
+        t17 = teams_el_15["team_name"] || teams_el_15[:team_name]
+        acc_13 += 1
+      end
+      t18 = acc_13
+      out << t18
+    end
+    out
+  end
+
+  def _avg_headcount_per_dept(input = @input)
+    out = []
+    t19 = input["departments"] || input[:departments]
+    t19.each_with_index do |departments_el_20, departments_i_21|
+      acc36 = 0
+      t37 = departments_el_20["teams"] || departments_el_20[:teams]
+      acc42 = 0
+      t37.each_with_index do |teams_el_6, teams_i_7|
+        t38 = teams_el_6["headcount"] || teams_el_6[:headcount]
+        acc36 += t38
+        t44 = teams_el_6["team_name"] || teams_el_6[:team_name]
+        acc42 += 1
+      end
+      t39 = acc36
+      t45 = acc42
+      t24 = t39 / t45.to_f
+      out << t24
+    end
+    out
+  end
+
+  def _is_above_average_team(input = @input)
+    out = []
+    t25 = input["departments"] || input[:departments]
+    t25.each_with_index do |departments_el_26, departments_i_27|
+      out_1 = []
+      t28 = departments_el_26["teams"] || departments_el_26[:teams]
+      acc52 = 0
+      acc58 = 0
+      t28.each_with_index do |teams_el_6, teams_i_7|
+        t54 = teams_el_6["headcount"] || teams_el_6[:headcount]
+        acc52 += t54
+        t60 = teams_el_6["team_name"] || teams_el_6[:team_name]
+        acc58 += 1
+      end
+      t55 = acc52
+      t61 = acc58
+      t49 = t55 / t61.to_f
+      t28.each_with_index do |teams_el_29, teams_i_30|
+        t31 = teams_el_29["headcount"] || teams_el_29[:headcount]
+        t33 = t31 > t49
+        out_1 << t33
+      end
+      out << out_1
+    end
+    out
+  end
+end

data/golden/reduction_broadcast/expected/snast.txt

@@ -0,0 +1,24 @@
+(SNAST
+  (VALUE dept_headcount
+    (Reduce :agg.sum over [teams]
+      (InputRef departments.dept.teams.team.headcount key_chain=[]) :: [departments, teams] -> integer
+    ) :: [departments] -> integer
+  ) :: [departments] -> integer
+  (VALUE teams_per_dept
+    (Reduce :agg.count over [teams]
+      (InputRef departments.dept.teams.team.team_name key_chain=[]) :: [departments, teams] -> string
+    ) :: [departments] -> integer
+  ) :: [departments] -> integer
+  (VALUE avg_headcount_per_dept
+    (Call :core.div
+      (Ref dept_headcount) :: [departments] -> integer
+      (Ref teams_per_dept) :: [departments] -> integer
+    ) :: [departments] -> float
+  ) :: [departments] -> float
+  (TRAIT is_above_average_team
+    (Call :core.gt
+      (InputRef departments.dept.teams.team.headcount key_chain=[]) :: [departments, teams] -> integer
+      (Ref avg_headcount_per_dept) :: [departments] -> float
+    ) :: [departments, teams] -> boolean
+  ) :: [departments, teams] -> boolean
+)

data/golden/reduction_broadcast/expected.json

@@ -0,0 +1,25 @@
+{
+    "dept_headcount": [
+      30,
+      40
+    ],
+    "teams_per_dept": [
+      3,
+      2
+    ],
+    "avg_headcount_per_dept": [
+      10.0,
+      20.0
+    ],
+    "is_above_average_team": [
+      [
+        false,
+        true,
+        false
+      ],
+      [
+        true,
+        false
+      ]
+    ]
+}

data/golden/reduction_broadcast/input.json

@@ -0,0 +1,19 @@
+{
+    "departments": [
+      {
+        "dept_name": "Engineering",
+        "teams": [
+          {"team_name": "Backend", "headcount": 10},
+          {"team_name": "Frontend", "headcount": 15},
+          {"team_name": "Data", "headcount": 5}
+        ]
+      },
+      {
+        "dept_name": "Sales",
+        "teams": [
+          {"team_name": "Enterprise", "headcount": 30},
+          {"team_name": "SMB", "headcount": 10}
+        ]
+      }
+    ]
+  }

data/golden/reduction_broadcast/schema.kumi

@@ -0,0 +1,34 @@
+schema do
+  input do
+    array :departments do
+      hash :dept do
+        string :dept_name
+        array :teams do
+          hash :team do
+            string :team_name
+            integer :headcount
+          end
+        end
+      end
+    end
+  end
+
+  # 1. Reduce to find the total headcount per department.
+  # Scope: [departments] -> integer
+  value :dept_headcount, fn(:sum, input.departments.dept.teams.team.headcount)
+
+  # 2. Count the number of teams per department.
+  # Scope: [departments] -> integer
+  value :teams_per_dept, fn(:count, input.departments.dept.teams.team.team_name)
+
+  # 3. Calculate the average headcount for each department.
+  # Note: This is a scalar value *at the department level*.
+  # Scope: [departments] -> float
+  value :avg_headcount_per_dept, dept_headcount / teams_per_dept
+
+  # 4. CORNER CASE: Broadcast the department-level average back down
+  # to the team level for comparison. The compiler must correctly align
+  # `avg_headcount_per_dept` with each team within that department.
+  # Scope: [departments, teams] -> boolean
+  trait :is_above_average_team, input.departments.dept.teams.team.headcount > avg_headcount_per_dept
+end

data/golden/roll/expected/ast.txt

@@ -0,0 +1,36 @@
+(Root
+  inputs: [
+    (InputDeclaration :cells :array
+      [
+        (InputDeclaration :value :integer)
+      ]
+    )
+  ]
+  values: [
+    (ValueDeclaration :roll_right
+      (CallExpression :roll
+        (InputElementReference cells.value)
+        (Literal 1)
+      )
+    )
+    (ValueDeclaration :roll_left
+      (CallExpression :roll
+        (InputElementReference cells.value)
+        (Literal -1)
+      )
+    )
+    (ValueDeclaration :roll_right_clamp
+      (CallExpression :roll
+        (InputElementReference cells.value)
+        (Literal 1)
+      )
+    )
+    (ValueDeclaration :roll_left_clamp
+      (CallExpression :roll
+        (InputElementReference cells.value)
+        (Literal -1)
+      )
+    )
+  ]
+  traits: []
+)

data/golden/roll/expected/input_plan.txt

@@ -0,0 +1,3 @@
+cells: array (array) access_mode=element
+  └─> value: element_access
+  value: integer

data/golden/roll/expected/lir_00_unoptimized.txt

@@ -0,0 +1,56 @@
+(LIR
+  (Declaration roll_right
+    %t1 = load_input "cells" :: array
+    loop cells id=L1 in %t1 as el=%cells_el_2, idx=%cells_i_3
+      %t4 = Length %t1 :: integer
+      %t5 = const 1 :: integer
+      %t6 = call core.sub(%cells_i_3, %t5) :: integer
+      %t7 = call core.mod(%t6, %t4) :: integer
+      %t8 = call core.add(%t7, %t4) :: integer
+      %t9 = call core.mod(%t8, %t4) :: integer
+      %t10 = Gather %t1, %t9 :: integer
+      yield %t10
+    end_loop
+  )
+  (Declaration roll_left
+    %t11 = load_input "cells" :: array
+    loop cells id=L2 in %t11 as el=%cells_el_12, idx=%cells_i_13
+      %t14 = Length %t11 :: integer
+      %t15 = const -1 :: integer
+      %t16 = call core.sub(%cells_i_13, %t15) :: integer
+      %t17 = call core.mod(%t16, %t14) :: integer
+      %t18 = call core.add(%t17, %t14) :: integer
+      %t19 = call core.mod(%t18, %t14) :: integer
+      %t20 = Gather %t11, %t19 :: integer
+      yield %t20
+    end_loop
+  )
+  (Declaration roll_right_clamp
+    %t21 = load_input "cells" :: array
+    loop cells id=L3 in %t21 as el=%cells_el_22, idx=%cells_i_23
+      %t24 = Length %t21 :: integer
+      %t25 = const 1 :: integer
+      %t26 = call core.sub(%cells_i_23, %t25) :: integer
+      %t27 = const 1 :: integer
+      %t28 = call core.sub(%t24, %t27) :: integer
+      %t29 = const 0 :: integer
+      %t30 = call core.clamp(%t26, %t29, %t28) :: integer
+      %t31 = Gather %t21, %t30 :: integer
+      yield %t31
+    end_loop
+  )
+  (Declaration roll_left_clamp
+    %t32 = load_input "cells" :: array
+    loop cells id=L4 in %t32 as el=%cells_el_33, idx=%cells_i_34
+      %t35 = Length %t32 :: integer
+      %t36 = const -1 :: integer
+      %t37 = call core.sub(%cells_i_34, %t36) :: integer
+      %t38 = const 1 :: integer
+      %t39 = call core.sub(%t35, %t38) :: integer
+      %t40 = const 0 :: integer
+      %t41 = call core.clamp(%t37, %t40, %t39) :: integer
+      %t42 = Gather %t32, %t41 :: integer
+      yield %t42
+    end_loop
+  )
+)

data/golden/roll/expected/lir_01_hoist_scalar_references.txt

@@ -0,0 +1,56 @@
+(LIR
+  (Declaration roll_right
+    %t1 = load_input "cells" :: array
+    loop cells id=L1 in %t1 as el=%cells_el_2, idx=%cells_i_3
+      %t4 = Length %t1 :: integer
+      %t5 = const 1 :: integer
+      %t6 = call core.sub(%cells_i_3, %t5) :: integer
+      %t7 = call core.mod(%t6, %t4) :: integer
+      %t8 = call core.add(%t7, %t4) :: integer
+      %t9 = call core.mod(%t8, %t4) :: integer
+      %t10 = Gather %t1, %t9 :: integer
+      yield %t10
+    end_loop
+  )
+  (Declaration roll_left
+    %t11 = load_input "cells" :: array
+    loop cells id=L2 in %t11 as el=%cells_el_12, idx=%cells_i_13
+      %t14 = Length %t11 :: integer
+      %t15 = const -1 :: integer
+      %t16 = call core.sub(%cells_i_13, %t15) :: integer
+      %t17 = call core.mod(%t16, %t14) :: integer
+      %t18 = call core.add(%t17, %t14) :: integer
+      %t19 = call core.mod(%t18, %t14) :: integer
+      %t20 = Gather %t11, %t19 :: integer
+      yield %t20
+    end_loop
+  )
+  (Declaration roll_right_clamp
+    %t21 = load_input "cells" :: array
+    loop cells id=L3 in %t21 as el=%cells_el_22, idx=%cells_i_23
+      %t24 = Length %t21 :: integer
+      %t25 = const 1 :: integer
+      %t26 = call core.sub(%cells_i_23, %t25) :: integer
+      %t27 = const 1 :: integer
+      %t28 = call core.sub(%t24, %t27) :: integer
+      %t29 = const 0 :: integer
+      %t30 = call core.clamp(%t26, %t29, %t28) :: integer
+      %t31 = Gather %t21, %t30 :: integer
+      yield %t31
+    end_loop
+  )
+  (Declaration roll_left_clamp
+    %t32 = load_input "cells" :: array
+    loop cells id=L4 in %t32 as el=%cells_el_33, idx=%cells_i_34
+      %t35 = Length %t32 :: integer
+      %t36 = const -1 :: integer
+      %t37 = call core.sub(%cells_i_34, %t36) :: integer
+      %t38 = const 1 :: integer
+      %t39 = call core.sub(%t35, %t38) :: integer
+      %t40 = const 0 :: integer
+      %t41 = call core.clamp(%t37, %t40, %t39) :: integer
+      %t42 = Gather %t32, %t41 :: integer
+      yield %t42
+    end_loop
+  )
+)

data/golden/roll/expected/lir_02_inlined.txt

@@ -0,0 +1,56 @@
+(LIR
+  (Declaration roll_right
+    %t1 = load_input "cells" :: array
+    loop cells id=L1 in %t1 as el=%cells_el_2, idx=%cells_i_3
+      %t4 = Length %t1 :: integer
+      %t5 = const 1 :: integer
+      %t6 = call core.sub(%cells_i_3, %t5) :: integer
+      %t7 = call core.mod(%t6, %t4) :: integer
+      %t8 = call core.add(%t7, %t4) :: integer
+      %t9 = call core.mod(%t8, %t4) :: integer
+      %t10 = Gather %t1, %t9 :: integer
+      yield %t10
+    end_loop
+  )
+  (Declaration roll_left
+    %t11 = load_input "cells" :: array
+    loop cells id=L2 in %t11 as el=%cells_el_12, idx=%cells_i_13
+      %t14 = Length %t11 :: integer
+      %t15 = const -1 :: integer
+      %t16 = call core.sub(%cells_i_13, %t15) :: integer
+      %t17 = call core.mod(%t16, %t14) :: integer
+      %t18 = call core.add(%t17, %t14) :: integer
+      %t19 = call core.mod(%t18, %t14) :: integer
+      %t20 = Gather %t11, %t19 :: integer
+      yield %t20
+    end_loop
+  )
+  (Declaration roll_right_clamp
+    %t21 = load_input "cells" :: array
+    loop cells id=L3 in %t21 as el=%cells_el_22, idx=%cells_i_23
+      %t24 = Length %t21 :: integer
+      %t25 = const 1 :: integer
+      %t26 = call core.sub(%cells_i_23, %t25) :: integer
+      %t27 = const 1 :: integer
+      %t28 = call core.sub(%t24, %t27) :: integer
+      %t29 = const 0 :: integer
+      %t30 = call core.clamp(%t26, %t29, %t28) :: integer
+      %t31 = Gather %t21, %t30 :: integer
+      yield %t31
+    end_loop
+  )
+  (Declaration roll_left_clamp
+    %t32 = load_input "cells" :: array
+    loop cells id=L4 in %t32 as el=%cells_el_33, idx=%cells_i_34
+      %t35 = Length %t32 :: integer
+      %t36 = const -1 :: integer
+      %t37 = call core.sub(%cells_i_34, %t36) :: integer
+      %t38 = const 1 :: integer
+      %t39 = call core.sub(%t35, %t38) :: integer
+      %t40 = const 0 :: integer
+      %t41 = call core.clamp(%t37, %t40, %t39) :: integer
+      %t42 = Gather %t32, %t41 :: integer
+      yield %t42
+    end_loop
+  )
+)

data/golden/roll/expected/lir_03_cse.txt

@@ -0,0 +1,55 @@
+(LIR
+  (Declaration roll_right
+    %t1 = load_input "cells" :: array
+    loop cells id=L1 in %t1 as el=%cells_el_2, idx=%cells_i_3
+      %t4 = Length %t1 :: integer
+      %t5 = const 1 :: integer
+      %t6 = call core.sub(%cells_i_3, %t5) :: integer
+      %t7 = call core.mod(%t6, %t4) :: integer
+      %t8 = call core.add(%t7, %t4) :: integer
+      %t9 = call core.mod(%t8, %t4) :: integer
+      %t10 = Gather %t1, %t9 :: integer
+      yield %t10
+    end_loop
+  )
+  (Declaration roll_left
+    %t11 = load_input "cells" :: array
+    loop cells id=L2 in %t11 as el=%cells_el_12, idx=%cells_i_13
+      %t14 = Length %t11 :: integer
+      %t15 = const -1 :: integer
+      %t16 = call core.sub(%cells_i_13, %t15) :: integer
+      %t17 = call core.mod(%t16, %t14) :: integer
+      %t18 = call core.add(%t17, %t14) :: integer
+      %t19 = call core.mod(%t18, %t14) :: integer
+      %t20 = Gather %t11, %t19 :: integer
+      yield %t20
+    end_loop
+  )
+  (Declaration roll_right_clamp
+    %t21 = load_input "cells" :: array
+    loop cells id=L3 in %t21 as el=%cells_el_22, idx=%cells_i_23
+      %t24 = Length %t21 :: integer
+      %t25 = const 1 :: integer
+      %t29 = const 0 :: integer
+      %t26 = call core.sub(%cells_i_23, %t25) :: integer
+      %t28 = call core.sub(%t24, %t25) :: integer
+      %t30 = call core.clamp(%t26, %t29, %t28) :: integer
+      %t31 = Gather %t21, %t30 :: integer
+      yield %t31
+    end_loop
+  )
+  (Declaration roll_left_clamp
+    %t32 = load_input "cells" :: array
+    loop cells id=L4 in %t32 as el=%cells_el_33, idx=%cells_i_34
+      %t35 = Length %t32 :: integer
+      %t36 = const -1 :: integer
+      %t38 = const 1 :: integer
+      %t40 = const 0 :: integer
+      %t37 = call core.sub(%cells_i_34, %t36) :: integer
+      %t39 = call core.sub(%t35, %t38) :: integer
+      %t41 = call core.clamp(%t37, %t40, %t39) :: integer
+      %t42 = Gather %t32, %t41 :: integer
+      yield %t42
+    end_loop
+  )
+)