kumi 0.0.24 → 0.0.25

data/golden/multirank_hoisting/expected/lir_04_1_loop_fusion.txt

@@ -29,19 +29,19 @@
     end_loop
   )
   (Declaration elem_affine
+    %t66 = load_input "global_offset" :: float
+    %t67 = const 1.0 :: float
+    %t68 = call core.add(%t66, %t67) :: float
     %t19 = load_input "batch" :: array
     loop batch id=L4 in %t19 as el=%batch_el_20, idx=%batch_i_21
+      %t64 = load_field batch_el_20["mean"] :: float
       %t22 = load_field batch_el_20["row"] :: any
-      %t81 = load_input "global_offset" :: float
-      %t82 = const 1.0 :: float
-      %t83 = call core.add(%t81, %t82) :: float
+      %t65 = call core.add(%t64, %t68) :: float
       loop row id=L5 in %t22 as el=%row_el_23, idx=%row_i_24
-        %t25 = load_field row_el_23["col"] :: any
         %t59 = load_field row_el_23["scale"] :: float
         %t60 = const 2.0 :: float
         %t61 = call core.mul(%t59, %t60) :: float
-        %t64 = load_field batch_el_20["mean"] :: float
-        %t65 = call core.add(%t64, %t83) :: float
+        %t25 = load_field row_el_23["col"] :: any
         loop col id=L6 in %t25 as el=%col_el_26, idx=%col_i_27
           %t28 = load_field col_el_26["val"] :: float
           %t30 = call core.mul(%t28, %t61) :: float
@@ -52,25 +52,25 @@
     end_loop
   )
   (Declaration row_sum_affine
+    %t86 = load_input "global_offset" :: float
+    %t87 = const 1.0 :: float
+    %t88 = call core.add(%t86, %t87) :: float
     %t33 = load_input "batch" :: array
     loop batch id=L7 in %t33 as el=%batch_el_34, idx=%batch_i_35
+      %t84 = load_field batch_el_34["mean"] :: float
       %t36 = load_field batch_el_34["row"] :: any
+      %t85 = call core.add(%t84, %t88) :: float
       loop row id=L8 in %t36 as el=%row_el_37, idx=%row_i_38
+        %t79 = load_field row_el_37["scale"] :: float
+        %t80 = const 2.0 :: float
+        %t81 = call core.mul(%t79, %t80) :: float
         %acc_39 = decl_acc :: float
         %t40 = load_field row_el_37["col"] :: any
-        %t86 = load_field row_el_37["scale"] :: float
-        %t87 = const 2.0 :: float
-        %t88 = call core.mul(%t86, %t87) :: float
-        %t90 = load_input "global_offset" :: float
-        %t91 = const 1.0 :: float
-        %t92 = call core.add(%t90, %t91) :: float
-        %t93 = load_field batch_el_34["mean"] :: float
-        %t94 = call core.add(%t93, %t92) :: float
         loop col id=L9 in %t40 as el=%col_el_41, idx=%col_i_42
-          %t69 = load_field col_el_41["val"] :: float
-          %t71 = call core.mul(%t69, %t88) :: float
-          %t73 = call core.add(%t71, %t94) :: float
-          %acc_39 = acc_add agg.sum(%acc_39, %t73) :: float
+          %t72 = load_field col_el_41["val"] :: float
+          %t74 = call core.mul(%t72, %t81) :: float
+          %t76 = call core.add(%t74, %t85) :: float
+          %acc_39 = acc_add agg.sum(%acc_39, %t76) :: float
         end_loop
         %t44 = acc_load %acc_39 :: float
         yield %t44
@@ -78,29 +78,29 @@
     end_loop
   )
   (Declaration batch_total_affine
+    %t115 = load_input "global_offset" :: float
+    %t116 = const 1.0 :: float
+    %t117 = call core.add(%t115, %t116) :: float
     %t45 = load_input "batch" :: array
     loop batch id=L10 in %t45 as el=%batch_el_46, idx=%batch_i_47
+      %t113 = load_field batch_el_46["mean"] :: float
       %acc_48 = decl_acc :: float
       %t49 = load_field batch_el_46["row"] :: any
+      %t114 = call core.add(%t113, %t117) :: float
       loop row id=L11 in %t49 as el=%row_el_50, idx=%row_i_51
-        %acc77 = decl_acc :: float
-        %t78 = load_field row_el_50["col"] :: any
-        %t110 = load_input "global_offset" :: float
-        %t111 = const 1.0 :: float
-        %t112 = call core.add(%t110, %t111) :: float
-        loop col id=L12 in %t78 as el=%col_el_41, idx=%col_i_42
-          %t100 = load_field row_el_50["scale"] :: float
-          %t101 = const 2.0 :: float
-          %t102 = call core.mul(%t100, %t101) :: float
-          %t105 = load_field batch_el_46["mean"] :: float
-          %t106 = call core.add(%t105, %t112) :: float
-          %t107 = load_field col_el_41["val"] :: float
-          %t108 = call core.mul(%t107, %t102) :: float
-          %t109 = call core.add(%t108, %t106) :: float
-          %acc77 = acc_add agg.sum(%acc77, %t109) :: float
+        %t108 = load_field row_el_50["scale"] :: float
+        %t109 = const 2.0 :: float
+        %t110 = call core.mul(%t108, %t109) :: float
+        %acc92 = decl_acc :: float
+        %t93 = load_field row_el_50["col"] :: any
+        loop col id=L12 in %t93 as el=%t94, idx=%t95
+          %t101 = load_field t94["val"] :: float
+          %t103 = call core.mul(%t101, %t110) :: float
+          %t105 = call core.add(%t103, %t114) :: float
+          %acc92 = acc_add agg.sum(%acc92, %t105) :: float
         end_loop
-        %t80 = acc_load %acc77 :: float
-        %acc_48 = acc_add agg.sum(%acc_48, %t80) :: float
+        %t97 = acc_load %acc92 :: float
+        %acc_48 = acc_add agg.sum(%acc_48, %t97) :: float
       end_loop
       %t53 = acc_load %acc_48 :: float
       yield %t53

data/golden/multirank_hoisting/expected/lir_04_loop_invcm.txt

@@ -29,19 +29,19 @@
     end_loop
   )
   (Declaration elem_affine
+    %t66 = load_input "global_offset" :: float
+    %t67 = const 1.0 :: float
+    %t68 = call core.add(%t66, %t67) :: float
     %t19 = load_input "batch" :: array
-    %t81 = load_input "global_offset" :: float
-    %t82 = const 1.0 :: float
     %t60 = const 2.0 :: float
-    %t83 = call core.add(%t81, %t82) :: float
     loop batch id=L4 in %t19 as el=%batch_el_20, idx=%batch_i_21
-      %t22 = load_field batch_el_20["row"] :: any
       %t64 = load_field batch_el_20["mean"] :: float
-      %t65 = call core.add(%t64, %t83) :: float
+      %t22 = load_field batch_el_20["row"] :: any
+      %t65 = call core.add(%t64, %t68) :: float
       loop row id=L5 in %t22 as el=%row_el_23, idx=%row_i_24
-        %t25 = load_field row_el_23["col"] :: any
         %t59 = load_field row_el_23["scale"] :: float
         %t61 = call core.mul(%t59, %t60) :: float
+        %t25 = load_field row_el_23["col"] :: any
         loop col id=L6 in %t25 as el=%col_el_26, idx=%col_i_27
           %t28 = load_field col_el_26["val"] :: float
           %t30 = call core.mul(%t28, %t61) :: float
@@ -52,25 +52,25 @@
     end_loop
   )
   (Declaration row_sum_affine
+    %t86 = load_input "global_offset" :: float
+    %t87 = const 1.0 :: float
+    %t88 = call core.add(%t86, %t87) :: float
     %t33 = load_input "batch" :: array
-    %t87 = const 2.0 :: float
-    %t90 = load_input "global_offset" :: float
-    %t91 = const 1.0 :: float
-    %t92 = call core.add(%t90, %t91) :: float
+    %t80 = const 2.0 :: float
     loop batch id=L7 in %t33 as el=%batch_el_34, idx=%batch_i_35
+      %t84 = load_field batch_el_34["mean"] :: float
       %t36 = load_field batch_el_34["row"] :: any
-      %t93 = load_field batch_el_34["mean"] :: float
-      %t94 = call core.add(%t93, %t92) :: float
+      %t85 = call core.add(%t84, %t88) :: float
       loop row id=L8 in %t36 as el=%row_el_37, idx=%row_i_38
+        %t79 = load_field row_el_37["scale"] :: float
+        %t81 = call core.mul(%t79, %t80) :: float
         %acc_39 = decl_acc :: float
         %t40 = load_field row_el_37["col"] :: any
-        %t86 = load_field row_el_37["scale"] :: float
-        %t88 = call core.mul(%t86, %t87) :: float
         loop col id=L9 in %t40 as el=%col_el_41, idx=%col_i_42
-          %t69 = load_field col_el_41["val"] :: float
-          %t71 = call core.mul(%t69, %t88) :: float
-          %t73 = call core.add(%t71, %t94) :: float
-          %acc_39 = acc_add agg.sum(%acc_39, %t73) :: float
+          %t72 = load_field col_el_41["val"] :: float
+          %t74 = call core.mul(%t72, %t81) :: float
+          %t76 = call core.add(%t74, %t85) :: float
+          %acc_39 = acc_add agg.sum(%acc_39, %t76) :: float
         end_loop
         %t44 = acc_load %acc_39 :: float
         yield %t44
@@ -78,29 +78,29 @@
     end_loop
   )
   (Declaration batch_total_affine
+    %t115 = load_input "global_offset" :: float
+    %t116 = const 1.0 :: float
+    %t117 = call core.add(%t115, %t116) :: float
     %t45 = load_input "batch" :: array
-    %t110 = load_input "global_offset" :: float
-    %t111 = const 1.0 :: float
-    %t101 = const 2.0 :: float
-    %t112 = call core.add(%t110, %t111) :: float
+    %t109 = const 2.0 :: float
     loop batch id=L10 in %t45 as el=%batch_el_46, idx=%batch_i_47
+      %t113 = load_field batch_el_46["mean"] :: float
       %acc_48 = decl_acc :: float
       %t49 = load_field batch_el_46["row"] :: any
-      %t105 = load_field batch_el_46["mean"] :: float
-      %t106 = call core.add(%t105, %t112) :: float
+      %t114 = call core.add(%t113, %t117) :: float
       loop row id=L11 in %t49 as el=%row_el_50, idx=%row_i_51
-        %acc77 = decl_acc :: float
-        %t78 = load_field row_el_50["col"] :: any
-        %t100 = load_field row_el_50["scale"] :: float
-        %t102 = call core.mul(%t100, %t101) :: float
-        loop col id=L12 in %t78 as el=%col_el_41, idx=%col_i_42
-          %t107 = load_field col_el_41["val"] :: float
-          %t108 = call core.mul(%t107, %t102) :: float
-          %t109 = call core.add(%t108, %t106) :: float
-          %acc77 = acc_add agg.sum(%acc77, %t109) :: float
+        %t108 = load_field row_el_50["scale"] :: float
+        %t110 = call core.mul(%t108, %t109) :: float
+        %acc92 = decl_acc :: float
+        %t93 = load_field row_el_50["col"] :: any
+        loop col id=L12 in %t93 as el=%t94, idx=%t95
+          %t101 = load_field t94["val"] :: float
+          %t103 = call core.mul(%t101, %t110) :: float
+          %t105 = call core.add(%t103, %t114) :: float
+          %acc92 = acc_add agg.sum(%acc92, %t105) :: float
         end_loop
-        %t80 = acc_load %acc77 :: float
-        %acc_48 = acc_add agg.sum(%acc_48, %t80) :: float
+        %t97 = acc_load %acc92 :: float
+        %acc_48 = acc_add agg.sum(%acc_48, %t97) :: float
       end_loop
       %t53 = acc_load %acc_48 :: float
       yield %t53

data/golden/multirank_hoisting/expected/lir_06_const_prop.txt

@@ -29,19 +29,19 @@
     end_loop
   )
   (Declaration elem_affine
+    %t66 = load_input "global_offset" :: float
+    %t67 = const 1.0 :: float
+    %t68 = call core.add(%t66, %t67) :: float
     %t19 = load_input "batch" :: array
-    %t81 = load_input "global_offset" :: float
-    %t82 = const 1.0 :: float
     %t60 = const 2.0 :: float
-    %t83 = call core.add(%t81, %t82) :: float
     loop batch id=L4 in %t19 as el=%batch_el_20, idx=%batch_i_21
-      %t22 = load_field batch_el_20["row"] :: any
       %t64 = load_field batch_el_20["mean"] :: float
-      %t65 = call core.add(%t64, %t83) :: float
+      %t22 = load_field batch_el_20["row"] :: any
+      %t65 = call core.add(%t64, %t68) :: float
       loop row id=L5 in %t22 as el=%row_el_23, idx=%row_i_24
-        %t25 = load_field row_el_23["col"] :: any
         %t59 = load_field row_el_23["scale"] :: float
         %t61 = call core.mul(%t59, %t60) :: float
+        %t25 = load_field row_el_23["col"] :: any
         loop col id=L6 in %t25 as el=%col_el_26, idx=%col_i_27
           %t28 = load_field col_el_26["val"] :: float
           %t30 = call core.mul(%t28, %t61) :: float
@@ -52,25 +52,25 @@
     end_loop
   )
   (Declaration row_sum_affine
+    %t86 = load_input "global_offset" :: float
+    %t87 = const 1.0 :: float
+    %t88 = call core.add(%t86, %t87) :: float
     %t33 = load_input "batch" :: array
-    %t87 = const 2.0 :: float
-    %t90 = load_input "global_offset" :: float
-    %t91 = const 1.0 :: float
-    %t92 = call core.add(%t90, %t91) :: float
+    %t80 = const 2.0 :: float
     loop batch id=L7 in %t33 as el=%batch_el_34, idx=%batch_i_35
+      %t84 = load_field batch_el_34["mean"] :: float
       %t36 = load_field batch_el_34["row"] :: any
-      %t93 = load_field batch_el_34["mean"] :: float
-      %t94 = call core.add(%t93, %t92) :: float
+      %t85 = call core.add(%t84, %t88) :: float
       loop row id=L8 in %t36 as el=%row_el_37, idx=%row_i_38
+        %t79 = load_field row_el_37["scale"] :: float
+        %t81 = call core.mul(%t79, %t80) :: float
         %acc_39 = decl_acc :: float
         %t40 = load_field row_el_37["col"] :: any
-        %t86 = load_field row_el_37["scale"] :: float
-        %t88 = call core.mul(%t86, %t87) :: float
         loop col id=L9 in %t40 as el=%col_el_41, idx=%col_i_42
-          %t69 = load_field col_el_41["val"] :: float
-          %t71 = call core.mul(%t69, %t88) :: float
-          %t73 = call core.add(%t71, %t94) :: float
-          %acc_39 = acc_add agg.sum(%acc_39, %t73) :: float
+          %t72 = load_field col_el_41["val"] :: float
+          %t74 = call core.mul(%t72, %t81) :: float
+          %t76 = call core.add(%t74, %t85) :: float
+          %acc_39 = acc_add agg.sum(%acc_39, %t76) :: float
         end_loop
         %t44 = acc_load %acc_39 :: float
         yield %t44
@@ -78,29 +78,29 @@
     end_loop
   )
   (Declaration batch_total_affine
+    %t115 = load_input "global_offset" :: float
+    %t116 = const 1.0 :: float
+    %t117 = call core.add(%t115, %t116) :: float
     %t45 = load_input "batch" :: array
-    %t110 = load_input "global_offset" :: float
-    %t111 = const 1.0 :: float
-    %t101 = const 2.0 :: float
-    %t112 = call core.add(%t110, %t111) :: float
+    %t109 = const 2.0 :: float
     loop batch id=L10 in %t45 as el=%batch_el_46, idx=%batch_i_47
+      %t113 = load_field batch_el_46["mean"] :: float
       %acc_48 = decl_acc :: float
       %t49 = load_field batch_el_46["row"] :: any
-      %t105 = load_field batch_el_46["mean"] :: float
-      %t106 = call core.add(%t105, %t112) :: float
+      %t114 = call core.add(%t113, %t117) :: float
       loop row id=L11 in %t49 as el=%row_el_50, idx=%row_i_51
-        %acc77 = decl_acc :: float
-        %t78 = load_field row_el_50["col"] :: any
-        %t100 = load_field row_el_50["scale"] :: float
-        %t102 = call core.mul(%t100, %t101) :: float
-        loop col id=L12 in %t78 as el=%col_el_41, idx=%col_i_42
-          %t107 = load_field col_el_41["val"] :: float
-          %t108 = call core.mul(%t107, %t102) :: float
-          %t109 = call core.add(%t108, %t106) :: float
-          %acc77 = acc_add agg.sum(%acc77, %t109) :: float
+        %t108 = load_field row_el_50["scale"] :: float
+        %t110 = call core.mul(%t108, %t109) :: float
+        %acc92 = decl_acc :: float
+        %t93 = load_field row_el_50["col"] :: any
+        loop col id=L12 in %t93 as el=%t94, idx=%t95
+          %t101 = load_field t94["val"] :: float
+          %t103 = call core.mul(%t101, %t110) :: float
+          %t105 = call core.add(%t103, %t114) :: float
+          %acc92 = acc_add agg.sum(%acc92, %t105) :: float
         end_loop
-        %t80 = acc_load %acc77 :: float
-        %acc_48 = acc_add agg.sum(%acc_48, %t80) :: float
+        %t97 = acc_load %acc92 :: float
+        %acc_48 = acc_add agg.sum(%acc_48, %t97) :: float
       end_loop
       %t53 = acc_load %acc_48 :: float
       yield %t53

data/golden/multirank_hoisting/expected/schema_javascript.mjs

@@ -38,21 +38,21 @@ export function _row_scale2(input) {
 
 export function _elem_affine(input) {
   let out = [];
+  let t66 = input["global_offset"];
+  const t67 = 1.0;
+  let t68 = t66 + t67;
   let t19 = input["batch"];
-  let t81 = input["global_offset"];
-  const t82 = 1.0;
   const t60 = 2.0;
-  let t83 = t81 + t82;
   t19.forEach((batch_el_20, batch_i_21) => {
     let out_1 = [];
-    let t22 = batch_el_20["row"];
     let t64 = batch_el_20["mean"];
-    let t65 = t64 + t83;
+    let t22 = batch_el_20["row"];
+    let t65 = t64 + t68;
     t22.forEach((row_el_23, row_i_24) => {
       let out_2 = [];
-      let t25 = row_el_23["col"];
       let t59 = row_el_23["scale"];
       let t61 = t59 * t60;
+      let t25 = row_el_23["col"];
       t25.forEach((col_el_26, col_i_27) => {
         let t28 = col_el_26["val"];
         let t30 = t28 * t61;
@@ -68,26 +68,26 @@ export function _elem_affine(input) {
 
 export function _row_sum_affine(input) {
   let out = [];
+  let t86 = input["global_offset"];
+  const t87 = 1.0;
+  let t88 = t86 + t87;
   let t33 = input["batch"];
-  const t87 = 2.0;
-  let t90 = input["global_offset"];
-  const t91 = 1.0;
-  let t92 = t90 + t91;
+  const t80 = 2.0;
   t33.forEach((batch_el_34, batch_i_35) => {
     let out_1 = [];
+    let t84 = batch_el_34["mean"];
     let t36 = batch_el_34["row"];
-    let t93 = batch_el_34["mean"];
-    let t94 = t93 + t92;
+    let t85 = t84 + t88;
     t36.forEach((row_el_37, row_i_38) => {
+      let t79 = row_el_37["scale"];
+      let t81 = t79 * t80;
       let acc_39 = 0.0;
       let t40 = row_el_37["col"];
-      let t86 = row_el_37["scale"];
-      let t88 = t86 * t87;
       t40.forEach((col_el_41, col_i_42) => {
-        let t69 = col_el_41["val"];
-        let t71 = t69 * t88;
-        let t73 = t71 + t94;
-        acc_39 += t73;
+        let t72 = col_el_41["val"];
+        let t74 = t72 * t81;
+        let t76 = t74 + t85;
+        acc_39 += t76;
       });
       out_1.push(acc_39);
     });
@@ -98,28 +98,28 @@ export function _row_sum_affine(input) {
 
 export function _batch_total_affine(input) {
   let out = [];
+  let t115 = input["global_offset"];
+  const t116 = 1.0;
+  let t117 = t115 + t116;
   let t45 = input["batch"];
-  let t110 = input["global_offset"];
-  const t111 = 1.0;
-  const t101 = 2.0;
-  let t112 = t110 + t111;
+  const t109 = 2.0;
   t45.forEach((batch_el_46, batch_i_47) => {
+    let t113 = batch_el_46["mean"];
     let acc_48 = 0.0;
     let t49 = batch_el_46["row"];
-    let t105 = batch_el_46["mean"];
-    let t106 = t105 + t112;
+    let t114 = t113 + t117;
     t49.forEach((row_el_50, row_i_51) => {
-      let acc77 = 0.0;
-      let t78 = row_el_50["col"];
-      let t100 = row_el_50["scale"];
-      let t102 = t100 * t101;
-      t78.forEach((col_el_41, col_i_42) => {
-        let t107 = col_el_41["val"];
-        let t108 = t107 * t102;
-        let t109 = t108 + t106;
-        acc77 += t109;
+      let t108 = row_el_50["scale"];
+      let t110 = t108 * t109;
+      let acc92 = 0.0;
+      let t93 = row_el_50["col"];
+      t93.forEach((t94, t95) => {
+        let t101 = t94["val"];
+        let t103 = t101 * t110;
+        let t105 = t103 + t114;
+        acc92 += t105;
       });
-      acc_48 += acc77;
+      acc_48 += acc92;
     });
     out.push(acc_48);
   });

data/golden/multirank_hoisting/expected/schema_ruby.rb

@@ -1,5 +1,5 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_e0419fbf641de528f8d1c03f124e177804f1a04ed69561ff0fe20f04192a818b
+module Kumi::Compiled::KUMI_7493f45c6a56b6958f928d8337e155ecf85850b56acef5a72dee8f1f697d1aac
   def self.from(input_data = nil)
     instance = Object.new
     instance.extend(self)
@@ -69,21 +69,21 @@ module Kumi::Compiled::KUMI_e0419fbf641de528f8d1c03f124e177804f1a04ed69561ff0fe2
 
   def _elem_affine(input = @input)
     out = []
+    t66 = input["global_offset"] || input[:global_offset]
+    t67 = 1.0
+    t68 = t66 + t67
     t19 = input["batch"] || input[:batch]
-    t81 = input["global_offset"] || input[:global_offset]
-    t82 = 1.0
     t60 = 2.0
-    t83 = t81 + t82
     t19.each_with_index do |batch_el_20, batch_i_21|
       out_1 = []
-      t22 = batch_el_20["row"] || batch_el_20[:row]
       t64 = batch_el_20["mean"] || batch_el_20[:mean]
-      t65 = t64 + t83
+      t22 = batch_el_20["row"] || batch_el_20[:row]
+      t65 = t64 + t68
       t22.each_with_index do |row_el_23, row_i_24|
         out_2 = []
-        t25 = row_el_23["col"] || row_el_23[:col]
         t59 = row_el_23["scale"] || row_el_23[:scale]
         t61 = t59 * t60
+        t25 = row_el_23["col"] || row_el_23[:col]
         t25.each_with_index do |col_el_26, col_i_27|
           t28 = col_el_26["val"] || col_el_26[:val]
           t30 = t28 * t61
@@ -99,26 +99,26 @@ module Kumi::Compiled::KUMI_e0419fbf641de528f8d1c03f124e177804f1a04ed69561ff0fe2
 
   def _row_sum_affine(input = @input)
     out = []
+    t86 = input["global_offset"] || input[:global_offset]
+    t87 = 1.0
+    t88 = t86 + t87
     t33 = input["batch"] || input[:batch]
-    t87 = 2.0
-    t90 = input["global_offset"] || input[:global_offset]
-    t91 = 1.0
-    t92 = t90 + t91
+    t80 = 2.0
     t33.each_with_index do |batch_el_34, batch_i_35|
       out_1 = []
+      t84 = batch_el_34["mean"] || batch_el_34[:mean]
       t36 = batch_el_34["row"] || batch_el_34[:row]
-      t93 = batch_el_34["mean"] || batch_el_34[:mean]
-      t94 = t93 + t92
+      t85 = t84 + t88
       t36.each_with_index do |row_el_37, row_i_38|
+        t79 = row_el_37["scale"] || row_el_37[:scale]
+        t81 = t79 * t80
         acc_39 = 0.0
         t40 = row_el_37["col"] || row_el_37[:col]
-        t86 = row_el_37["scale"] || row_el_37[:scale]
-        t88 = t86 * t87
         t40.each_with_index do |col_el_41, col_i_42|
-          t69 = col_el_41["val"] || col_el_41[:val]
-          t71 = t69 * t88
-          t73 = t71 + t94
-          acc_39 += t73
+          t72 = col_el_41["val"] || col_el_41[:val]
+          t74 = t72 * t81
+          t76 = t74 + t85
+          acc_39 += t76
         end
         t44 = acc_39
         out_1 << t44
@@ -130,29 +130,29 @@ module Kumi::Compiled::KUMI_e0419fbf641de528f8d1c03f124e177804f1a04ed69561ff0fe2
 
   def _batch_total_affine(input = @input)
     out = []
+    t115 = input["global_offset"] || input[:global_offset]
+    t116 = 1.0
+    t117 = t115 + t116
     t45 = input["batch"] || input[:batch]
-    t110 = input["global_offset"] || input[:global_offset]
-    t111 = 1.0
-    t101 = 2.0
-    t112 = t110 + t111
+    t109 = 2.0
     t45.each_with_index do |batch_el_46, batch_i_47|
+      t113 = batch_el_46["mean"] || batch_el_46[:mean]
       acc_48 = 0.0
       t49 = batch_el_46["row"] || batch_el_46[:row]
-      t105 = batch_el_46["mean"] || batch_el_46[:mean]
-      t106 = t105 + t112
+      t114 = t113 + t117
       t49.each_with_index do |row_el_50, row_i_51|
-        acc77 = 0.0
-        t78 = row_el_50["col"] || row_el_50[:col]
-        t100 = row_el_50["scale"] || row_el_50[:scale]
-        t102 = t100 * t101
-        t78.each_with_index do |col_el_41, col_i_42|
-          t107 = col_el_41["val"] || col_el_41[:val]
-          t108 = t107 * t102
-          t109 = t108 + t106
-          acc77 += t109
+        t108 = row_el_50["scale"] || row_el_50[:scale]
+        t110 = t108 * t109
+        acc92 = 0.0
+        t93 = row_el_50["col"] || row_el_50[:col]
+        t93.each_with_index do |t94, t95|
+          t101 = t94["val"] || t94[:val]
+          t103 = t101 * t110
+          t105 = t103 + t114
+          acc92 += t105
         end
-        t80 = acc77
-        acc_48 += t80
+        t97 = acc92
+        acc_48 += t97
       end
       t53 = acc_48
       out << t53

data/golden/nested_hash/expected/schema_ruby.rb

@@ -1,5 +1,5 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_3b954603dedc568d2f576b3345109d43555583c88706a3dbe86f8e07cbcfdeb4
+module Kumi::Compiled::KUMI_06206c81b13d30ba2dfe33471a73d550acd8ceaaed39c3bc060cd965ef719e19
   def self.from(input_data = nil)
     instance = Object.new
     instance.extend(self)

data/golden/reduction_broadcast/expected/lir_02_inlined.txt

@@ -31,48 +31,48 @@
       %t34 = load_input "departments" :: array
       %acc36 = decl_acc :: integer
       %t37 = load_field departments_el_20["teams"] :: any
-      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
+      loop teams id=L8 in %t37 as el=%t38, idx=%t39
+        %t40 = load_field t38["headcount"] :: integer
+        %acc36 = acc_add agg.sum(%acc36, %t40) :: integer
       end_loop
-      %t39 = acc_load %acc36 :: integer
-      %t40 = load_input "departments" :: array
-      %acc42 = decl_acc :: integer
-      %t43 = load_field departments_el_20["teams"] :: any
-      loop teams id=L9 in %t43 as el=%teams_el_15, idx=%teams_i_16
-        %t44 = load_field teams_el_15["team_name"] :: string
-        %acc42 = acc_add agg.count(%acc42, %t44) :: integer
+      %t41 = acc_load %acc36 :: integer
+      %t42 = load_input "departments" :: array
+      %acc44 = decl_acc :: integer
+      %t45 = load_field departments_el_20["teams"] :: any
+      loop teams id=L9 in %t45 as el=%t46, idx=%t47
+        %t48 = load_field t46["team_name"] :: string
+        %acc44 = acc_add agg.count(%acc44, %t48) :: integer
       end_loop
-      %t45 = acc_load %acc42 :: integer
-      %t24 = call core.div(%t39, %t45) :: float
+      %t49 = acc_load %acc44 :: integer
+      %t24 = call core.div(%t41, %t49) :: 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
-      %t46 = load_input "departments" :: array
-      %t50 = load_input "departments" :: array
-      %acc52 = decl_acc :: integer
-      %t53 = load_field departments_el_26["teams"] :: any
-      loop teams id=L10 in %t53 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
+      %t54 = load_input "departments" :: array
+      %acc56 = decl_acc :: integer
+      %t57 = load_field departments_el_26["teams"] :: any
+      loop teams id=L10 in %t57 as el=%t58, idx=%t59
+        %t60 = load_field t58["headcount"] :: integer
+        %acc56 = acc_add agg.sum(%acc56, %t60) :: integer
       end_loop
-      %t55 = acc_load %acc52 :: integer
-      %t56 = load_input "departments" :: array
-      %acc58 = decl_acc :: integer
-      %t59 = load_field departments_el_26["teams"] :: any
-      loop teams id=L11 in %t59 as el=%teams_el_15, idx=%teams_i_16
-        %t60 = load_field teams_el_15["team_name"] :: string
-        %acc58 = acc_add agg.count(%acc58, %t60) :: integer
+      %t61 = acc_load %acc56 :: integer
+      %t62 = load_input "departments" :: array
+      %acc64 = decl_acc :: integer
+      %t65 = load_field departments_el_26["teams"] :: any
+      loop teams id=L11 in %t65 as el=%t66, idx=%t67
+        %t68 = load_field t66["team_name"] :: string
+        %acc64 = acc_add agg.count(%acc64, %t68) :: integer
       end_loop
-      %t61 = acc_load %acc58 :: integer
-      %t49 = call core.div(%t55, %t61) :: float
+      %t69 = acc_load %acc64 :: integer
+      %t50 = load_input "departments" :: array
+      %t53 = call core.div(%t61, %t69) :: float
+      %t28 = load_field departments_el_26["teams"] :: any
       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
+        %t33 = call core.gt(%t31, %t53) :: boolean
         yield %t33
       end_loop
     end_loop