kumi 0.0.31 → 0.0.33

data/golden/multirank_hoisting/expected/schema_ruby.rb

@@ -1,47 +1,15 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582b8de5c8f192c
-  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 :global_offset_plus then _global_offset_plus
-    when :batch_bias then _batch_bias
-    when :row_scale2 then _row_scale2
-    when :elem_affine then _elem_affine
-    when :row_sum_affine then _row_sum_affine
-    when :batch_total_affine then _batch_total_affine
-    else raise KeyError, "Unknown declaration"
-    end
-  end
-
-  def _global_offset_plus(input = @input)
+module Kumi::Compiled::KUMI_370ac782feb059e263a962e9d2c6b5de9914ae3e7636c3efdc00616eff6659b5
+  def self._global_offset_plus(input)
     t1 = input["global_offset"] || input[:global_offset]
-    t2 = 1.0
-    t1 + t2
+    t1 + 1.0
   end
 
-  def _batch_bias(input = @input)
+  def self._batch_bias(input)
     out = []
     t4 = input["batch"] || input[:batch]
     t54 = input["global_offset"] || input[:global_offset]
-    t55 = 1.0
-    t56 = t54 + t55
+    t56 = t54 + 1.0
     t4.each_with_index do |batch_el_5, batch_i_6|
       t7 = batch_el_5["mean"] || batch_el_5[:mean]
       t9 = t7 + t56
@@ -50,16 +18,15 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
     out
   end
 
-  def _row_scale2(input = @input)
+  def self._row_scale2(input)
     out = []
     t10 = input["batch"] || input[:batch]
-    t17 = 2.0
     t10.each_with_index do |batch_el_11, batch_i_12|
       out_1 = []
       t13 = batch_el_11["row"] || batch_el_11[:row]
       t13.each_with_index do |row_el_14, row_i_15|
         t16 = row_el_14["scale"] || row_el_14[:scale]
-        t18 = t16 * t17
+        t18 = t16 * 2.0
         out_1 << t18
       end
       out << out_1
@@ -67,13 +34,11 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
     out
   end
 
-  def _elem_affine(input = @input)
+  def self._elem_affine(input)
     out = []
     t66 = input["global_offset"] || input[:global_offset]
-    t67 = 1.0
-    t68 = t66 + t67
+    t68 = t66 + 1.0
     t19 = input["batch"] || input[:batch]
-    t60 = 2.0
     t19.each_with_index do |batch_el_20, batch_i_21|
       out_1 = []
       t64 = batch_el_20["mean"] || batch_el_20[:mean]
@@ -82,7 +47,7 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
       t22.each_with_index do |row_el_23, row_i_24|
         out_2 = []
         t59 = row_el_23["scale"] || row_el_23[:scale]
-        t61 = t59 * t60
+        t61 = t59 * 2.0
         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]
@@ -97,13 +62,11 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
     out
   end
 
-  def _row_sum_affine(input = @input)
+  def self._row_sum_affine(input)
     out = []
     t86 = input["global_offset"] || input[:global_offset]
-    t87 = 1.0
-    t88 = t86 + t87
+    t88 = t86 + 1.0
     t33 = input["batch"] || input[:batch]
-    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]
@@ -111,7 +74,7 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
       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
+        t81 = t79 * 2.0
         acc_39 = 0.0
         t40 = row_el_37["col"] || row_el_37[:col]
         t40.each_with_index do |col_el_41, col_i_42|
@@ -128,13 +91,11 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
     out
   end
 
-  def _batch_total_affine(input = @input)
+  def self._batch_total_affine(input)
     out = []
     t115 = input["global_offset"] || input[:global_offset]
-    t116 = 1.0
-    t117 = t115 + t116
+    t117 = t115 + 1.0
     t45 = input["batch"] || input[:batch]
-    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
@@ -142,7 +103,7 @@ module Kumi::Compiled::KUMI_38e52950a42d13f281d275be711f674d914feaf059f72c64a582
       t114 = t113 + t117
       t49.each_with_index do |row_el_50, row_i_51|
         t108 = row_el_50["scale"] || row_el_50[:scale]
-        t110 = t108 * t109
+        t110 = t108 * 2.0
         acc92 = 0.0
         t93 = row_el_50["col"] || row_el_50[:col]
         t93.each_with_index do |t94, t95|

data/golden/nested_hash/expected/lir_06_const_prop.txt

@@ -3,8 +3,7 @@
     %t1 = load_input "x" :: any
     %t2 = load_field t1["y"] :: any
     %t3 = load_field t2["z"] :: integer
-    %t4 = const 2 :: integer
-    %t5 = call core.mul:numeric(%t3, %t4) :: integer
+    %t5 = call core.mul:numeric(%t3, 2) :: integer
     yield %t5
   )
 )

data/golden/nested_hash/expected/schema_javascript.mjs

@@ -2,8 +2,7 @@ export function _double(input) {
   let t1 = input["x"];
   let t2 = t1["y"];
   let t3 = t2["z"];
-  const t4 = 2;
-  let t5 = t3 * t4;
+  let t5 = t3 * 2;
   return t5;
 }
 

data/golden/nested_hash/expected/schema_ruby.rb

@@ -1,35 +1,9 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_968c950e9550ea6866e905fc32ac2ae9aee50f3d8316a0580fb2ed0efb31b47b
-  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 :double then _double
-    else raise KeyError, "Unknown declaration"
-    end
-  end
-
-  def _double(input = @input)
+module Kumi::Compiled::KUMI_828578820f1f1a58e1e2eeaad09f06b89503ea9f58ca364a2b5d2440cb379ec1
+  def self._double(input)
     t1 = input["x"] || input[:x]
     t2 = t1["y"] || t1[:y]
     t3 = t2["z"] || t2[:z]
-    t4 = 2
-    t3 * t4
+    t3 * 2
   end
 end

data/golden/reduction_broadcast/expected/schema_ruby.rb

@@ -1,34 +1,6 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_4d7e7d13cea88587529ab5ca02b7786b61419b3ed17afe69d36b2d1947311324
-  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)
+module Kumi::Compiled::KUMI_735e9f0503a157662ecacc10e3a875ed90b8874700963efbf20ba6199fe41a3f
+  def self._dept_headcount(input)
     out = []
     t1 = input["departments"] || input[:departments]
     t1.each_with_index do |departments_el_2, departments_i_3|
@@ -44,7 +16,7 @@ module Kumi::Compiled::KUMI_4d7e7d13cea88587529ab5ca02b7786b61419b3ed17afe69d36b
     out
   end
 
-  def _teams_per_dept(input = @input)
+  def self._teams_per_dept(input)
     out = []
     t10 = input["departments"] || input[:departments]
     t10.each_with_index do |departments_el_11, departments_i_12|
@@ -60,7 +32,7 @@ module Kumi::Compiled::KUMI_4d7e7d13cea88587529ab5ca02b7786b61419b3ed17afe69d36b
     out
   end
 
-  def _avg_headcount_per_dept(input = @input)
+  def self._avg_headcount_per_dept(input)
     out = []
     t19 = input["departments"] || input[:departments]
     t19.each_with_index do |departments_el_20, departments_i_21|
@@ -81,7 +53,7 @@ module Kumi::Compiled::KUMI_4d7e7d13cea88587529ab5ca02b7786b61419b3ed17afe69d36b
     out
   end
 
-  def _is_above_average_team(input = @input)
+  def self._is_above_average_team(input)
     out = []
     t25 = input["departments"] || input[:departments]
     t25.each_with_index do |departments_el_26, departments_i_27|

data/golden/roll/expected/lir_06_const_prop.txt

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

data/golden/roll/expected/schema_javascript.mjs

@@ -2,9 +2,8 @@ export function _roll_right(input) {
   let out = [];
   let t1 = input["cells"];
   let t4 = t1.length;
-  const t5 = 1;
   t1.forEach((cells_el_2, cells_i_3) => {
-    let t6 = cells_i_3 - t5;
+    let t6 = cells_i_3 - 1;
     let t7 = ((t6 % t4) + t4) % t4;
     let t8 = t7 + t4;
     let t9 = ((t8 % t4) + t4) % t4;
@@ -18,9 +17,8 @@ export function _roll_left(input) {
   let out = [];
   let t11 = input["cells"];
   let t14 = t11.length;
-  const t15 = -1;
   t11.forEach((cells_el_12, cells_i_13) => {
-    let t16 = cells_i_13 - t15;
+    let t16 = cells_i_13 - -1;
     let t17 = ((t16 % t14) + t14) % t14;
     let t18 = t17 + t14;
     let t19 = ((t18 % t14) + t14) % t14;
@@ -34,12 +32,10 @@ export function _roll_right_clamp(input) {
   let out = [];
   let t21 = input["cells"];
   let t24 = t21.length;
-  const t25 = 1;
-  const t29 = 0;
-  let t28 = t24 - t25;
+  let t28 = t24 - 1;
   t21.forEach((cells_el_22, cells_i_23) => {
-    let t26 = cells_i_23 - t25;
-    let t30 = Math.min(Math.max(t26, t29), t28);
+    let t26 = cells_i_23 - 1;
+    let t30 = Math.min(Math.max(t26, 0), t28);
     let t31 = t21[t30];
     out.push(t31);
   });
@@ -50,13 +46,10 @@ export function _roll_left_clamp(input) {
   let out = [];
   let t32 = input["cells"];
   let t35 = t32.length;
-  const t36 = -1;
-  const t38 = 1;
-  const t40 = 0;
-  let t39 = t35 - t38;
+  let t39 = t35 - 1;
   t32.forEach((cells_el_33, cells_i_34) => {
-    let t37 = cells_i_34 - t36;
-    let t41 = Math.min(Math.max(t37, t40), t39);
+    let t37 = cells_i_34 - -1;
+    let t41 = Math.min(Math.max(t37, 0), t39);
     let t42 = t32[t41];
     out.push(t42);
   });

data/golden/roll/expected/schema_ruby.rb

@@ -1,40 +1,11 @@
 # Autogenerated by Kumi Codegen
-module Kumi::Compiled::KUMI_a1445291f04923fe3d936c1514d8bf48488a469a9328b03080feb08b70093276
-  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 :roll_right then _roll_right
-    when :roll_left then _roll_left
-    when :roll_right_clamp then _roll_right_clamp
-    when :roll_left_clamp then _roll_left_clamp
-    else raise KeyError, "Unknown declaration"
-    end
-  end
-
-  def _roll_right(input = @input)
+module Kumi::Compiled::KUMI_5b73d82c56cb65fcb0e02e2b26a3f7f674fc645882bff7e29b0e469f05a743c1
+  def self._roll_right(input)
     out = []
     t1 = input["cells"] || input[:cells]
     t4 = t1.length
-    t5 = 1
     t1.each_with_index do |cells_el_2, cells_i_3|
-      t6 = cells_i_3 - t5
+      t6 = cells_i_3 - 1
       t7 = t6 % t4
       t8 = t7 + t4
       t9 = t8 % t4
@@ -44,13 +15,12 @@ module Kumi::Compiled::KUMI_a1445291f04923fe3d936c1514d8bf48488a469a9328b03080fe
     out
   end
 
-  def _roll_left(input = @input)
+  def self._roll_left(input)
     out = []
     t11 = input["cells"] || input[:cells]
     t14 = t11.length
-    t15 = -1
     t11.each_with_index do |cells_el_12, cells_i_13|
-      t16 = cells_i_13 - t15
+      t16 = cells_i_13 - -1
       t17 = t16 % t14
       t18 = t17 + t14
       t19 = t18 % t14
@@ -60,33 +30,28 @@ module Kumi::Compiled::KUMI_a1445291f04923fe3d936c1514d8bf48488a469a9328b03080fe
     out
   end
 
-  def _roll_right_clamp(input = @input)
+  def self._roll_right_clamp(input)
     out = []
     t21 = input["cells"] || input[:cells]
     t24 = t21.length
-    t25 = 1
-    t29 = 0
-    t28 = t24 - t25
+    t28 = t24 - 1
     t21.each_with_index do |cells_el_22, cells_i_23|
-      t26 = cells_i_23 - t25
-      t30 = [[ t26, t29 ].max, t28 ].min
+      t26 = cells_i_23 - 1
+      t30 = t26.clamp(0, t28)
       t31 = t21[t30]
       out << t31
     end
     out
   end
 
-  def _roll_left_clamp(input = @input)
+  def self._roll_left_clamp(input)
     out = []
     t32 = input["cells"] || input[:cells]
     t35 = t32.length
-    t36 = -1
-    t38 = 1
-    t40 = 0
-    t39 = t35 - t38
+    t39 = t35 - 1
     t32.each_with_index do |cells_el_33, cells_i_34|
-      t37 = cells_i_34 - t36
-      t41 = [[ t37, t40 ].max, t39 ].min
+      t37 = cells_i_34 - -1
+      t41 = t37.clamp(0, t39)
       t42 = t32[t41]
       out << t42
     end

data/golden/schema_imports_broadcasting_with_imports/expected/ast.txt

@@ -0,0 +1,26 @@
+(Root
+  inputs: [
+    (InputDeclaration :items :array
+      [
+        (InputDeclaration :item :hash
+          [
+            (InputDeclaration :amount :decimal)
+          ]
+        )
+      ]
+    )
+  ]
+  values: [
+    (ValueDeclaration :item_taxes
+      (ImportCall :tax
+        (amount (InputElementReference items.item.amount))
+      )
+    )
+    (ValueDeclaration :total_tax
+      (CallExpression :sum
+        (DeclarationReference :item_taxes)
+      )
+    )
+  ]
+  traits: []
+)

data/golden/schema_imports_broadcasting_with_imports/expected/input_plan.txt

@@ -0,0 +1,5 @@
+items: array (array) access_mode=element
+  └─> item: element_access
+  item: hash (hash)
+    └─> amount: property_access
+    amount: decimal

data/golden/schema_imports_broadcasting_with_imports/expected/lir_00_unoptimized.txt

@@ -0,0 +1,20 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_decl item_taxes axes=[items] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t10) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_01_hoist_scalar_references.txt

@@ -0,0 +1,20 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t10 = load_decl item_taxes axes=[items] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t10) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_02_inlined.txt

@@ -0,0 +1,22 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t12 = load_input "items" :: array
+      %t13 = load_field items_el_8["amount"] :: decimal
+      %t14 = importschemacall %t13 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t14) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_03_cse.txt

@@ -0,0 +1,21 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t13 = load_field items_el_8["amount"] :: decimal
+      %t14 = importschemacall %t13 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t14) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_04_1_loop_fusion.txt

@@ -0,0 +1,21 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t13 = load_field items_el_8["amount"] :: decimal
+      %t14 = importschemacall %t13 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t14) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_04_loop_invcm.txt

@@ -0,0 +1,21 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t13 = load_field items_el_8["amount"] :: decimal
+      %t14 = importschemacall %t13 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t14) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/lir_06_const_prop.txt

@@ -0,0 +1,21 @@
+(LIR
+  (Declaration item_taxes
+    %t1 = load_input "items" :: array
+    loop items id=L1 in %t1 as el=%items_el_2, idx=%items_i_3
+      %t4 = load_field items_el_2["amount"] :: decimal
+      %t5 = importschemacall %t4 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      yield %t5
+    end_loop
+  )
+  (Declaration total_tax
+    %acc_6 = decl_acc :: float
+    %t7 = load_input "items" :: array
+    loop items id=L2 in %t7 as el=%items_el_8, idx=%items_i_9
+      %t13 = load_field items_el_8["amount"] :: decimal
+      %t14 = importschemacall %t13 fn_name="tax" source_module="Kumi::TestSharedSchemas::Tax" input_mapping_keys=["amount"] :: float
+      %acc_6 = acc_add agg.sum(%acc_6, %t14) :: float
+    end_loop
+    %t11 = acc_load %acc_6 :: float
+    yield %t11
+  )
+)

data/golden/schema_imports_broadcasting_with_imports/expected/nast.txt

@@ -0,0 +1,12 @@
+(NAST
+  (VALUE item_taxes
+    (ImportCall :tax
+    (amount (InputRef [:items, :item, :amount]))
+  )
+  )
+  (VALUE total_tax
+    (Call :sum
+    (Ref item_taxes)
+  )
+  )
+)