kumi 0.0.17 → 0.0.19

data/golden/array_element/expected/ast.txt

@@ -0,0 +1,21 @@
+(Root
+  inputs: [
+    (InputDeclaration :x :array
+      [
+        (InputDeclaration :y :array
+          [
+            (InputDeclaration :v :integer)
+          ]
+        )
+      ]
+    )
+  ]
+  values: [
+    (ValueDeclaration :v_sum_y
+      (CallExpression :sum
+        (InputElementReference x.y.v)
+      )
+    )
+  ]
+  traits: []
+)

data/golden/array_element/expected/input_plan.txt

@@ -0,0 +1,5 @@
+x: array (array) access_mode=element
+  └─> y: element_access → array_loop
+  y: array (array) access_mode=element
+    └─> v: element_access
+    v: integer

data/golden/array_element/expected/lir_00_unoptimized.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_01_hoist_scalar_references.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_02_inlined.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_03_cse.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_04_1_loop_fusion.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_04_loop_invcm.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/lir_06_const_prop.txt

@@ -0,0 +1,13 @@
+(LIR
+  (Declaration v_sum_y
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %acc_4 = decl_acc :: integer
+      loop y id=L2 in %x_el_2 as el=%y_el_5, idx=%y_i_6
+        %acc_4 = acc_add agg.sum(%acc_4, %y_el_5) :: integer
+      end_loop
+      %t7 = acc_load %acc_4 :: integer
+      yield %t7
+    end_loop
+  )
+)

data/golden/array_element/expected/nast.txt

@@ -0,0 +1,7 @@
+(NAST
+  (VALUE v_sum_y
+    (Call :"agg.sum"
+    (InputRef [:x, :y, :v])
+  )
+  )
+)

data/golden/array_element/expected/schema_javascript.mjs

@@ -0,0 +1,13 @@
+export function _v_sum_y(input) {
+  let out = [];
+  let t1 = input["x"];
+  t1.forEach((x_el_2, x_i_3) => {
+    let acc_4 = 0;
+    x_el_2.forEach((y_el_5, y_i_6) => {
+      acc_4 += y_el_5;
+    });
+    out.push(acc_4);
+  });
+  return out;
+}
+

data/golden/array_element/expected/schema_ruby.rb

@@ -0,0 +1,41 @@
+# Autogenerated by Kumi Codegen
+module Kumi::Compiled::KUMI_1412b74cf62d9ac249af9fd939b136c4f77dc87035a46bf635f2e5274454e431
+  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 :v_sum_y then _v_sum_y
+    else raise KeyError, "Unknown declaration"
+    end
+  end
+
+  def _v_sum_y(input = @input)
+    out = []
+    t1 = input["x"] || input[:x]
+    t1.each_with_index do |x_el_2, x_i_3|
+      acc_4 = 0
+      x_el_2.each_with_index do |y_el_5, y_i_6|
+        acc_4 += y_el_5
+      end
+      t7 = acc_4
+      out << t7
+    end
+    out
+  end
+end

data/golden/array_element/expected/snast.txt

@@ -0,0 +1,7 @@
+(SNAST
+  (VALUE v_sum_y
+    (Reduce :agg.sum over [y]
+      (InputRef x.y.v key_chain=[]) :: [x, y] -> integer
+    ) :: [x] -> integer
+  ) :: [x] -> integer
+)

data/golden/array_element/expected.json

@@ -0,0 +1,5 @@
+{
+    "v_sum_y": [
+        6
+    ]
+}

data/golden/array_element/input.json

@@ -0,0 +1,9 @@
+{
+    "x": [
+        [
+            1,
+            2,
+            3
+        ]
+    ]
+}

data/golden/array_element/schema.kumi

@@ -0,0 +1,11 @@
+schema do
+  input do
+    array :x do
+      array :y do
+        integer :v
+      end
+    end
+  end
+
+  value :v_sum_y, fn(:sum, input.x.y.v)
+end

data/golden/array_index/expected/ast.txt

@@ -0,0 +1,59 @@
+(Root
+  inputs: [
+    (InputDeclaration :x :array :i
+      [
+        (InputDeclaration :y :array :j
+          [
+            (InputDeclaration :_ :integer)
+          ]
+        )
+      ]
+    )
+  ]
+  values: [
+    (ValueDeclaration :W
+      (CallExpression :array_size
+        (InputElementReference x.y)
+      )
+    )
+    (ValueDeclaration :box
+      (CallExpression :add
+        (CallExpression :multiply
+          (CallExpression :index
+            (Literal :i)
+          )
+          (DeclarationReference :W)
+        )
+        (CallExpression :index
+          (Literal :j)
+        )
+      )
+    )
+    (ValueDeclaration :col_major
+      (CallExpression :add
+        (CallExpression :multiply
+          (CallExpression :index
+            (Literal :j)
+          )
+          (CallExpression :array_size
+            (InputReference :x)
+          )
+        )
+        (CallExpression :index
+          (Literal :i)
+        )
+      )
+    )
+    (ValueDeclaration :sum_ij
+      (CallExpression :add
+        (CallExpression :index
+          (Literal :i)
+        )
+        (CallExpression :index
+          (Literal :j)
+        )
+      )
+    )
+  ]
+  traits: []
+)

data/golden/array_index/expected/input_plan.txt

@@ -0,0 +1,5 @@
+x: array (array) access_mode=element
+  └─> y: element_access → array_loop
+  y: array (array) access_mode=element
+    └─> _: element_access
+    _: integer

data/golden/array_index/expected/lir_00_unoptimized.txt

@@ -0,0 +1,41 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t10 = load_decl W axes=[x] :: integer
+        %t11 = call core.mul(%x_i_7, %t10) :: integer
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t18 = load_input "x" :: array
+        %t19 = call core.array_size(%t18) :: integer
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_01_hoist_scalar_references.txt

@@ -0,0 +1,41 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t10 = load_decl W axes=[x] :: integer
+        %t11 = call core.mul(%x_i_7, %t10) :: integer
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t18 = load_input "x" :: array
+        %t19 = call core.array_size(%t18) :: integer
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_02_inlined.txt

@@ -0,0 +1,42 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      %t28 = load_input "x" :: array
+      %t29 = call core.array_size(%x_el_6) :: integer
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t11 = call core.mul(%x_i_7, %t29) :: integer
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t18 = load_input "x" :: array
+        %t19 = call core.array_size(%t18) :: integer
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_03_cse.txt

@@ -0,0 +1,40 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      %t29 = call core.array_size(%x_el_6) :: integer
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t11 = call core.mul(%x_i_7, %t29) :: integer
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t19 = call core.array_size(%t13) :: integer
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_04_1_loop_fusion.txt

@@ -0,0 +1,40 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      %t29 = call core.array_size(%x_el_6) :: integer
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t11 = call core.mul(%x_i_7, %t29) :: integer
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t19 = call core.array_size(%t13) :: integer
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_04_loop_invcm.txt

@@ -0,0 +1,40 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      %t29 = call core.array_size(%x_el_6) :: integer
+      %t11 = call core.mul(%x_i_7, %t29) :: integer
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    %t19 = call core.array_size(%t13) :: integer
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/lir_06_const_prop.txt

@@ -0,0 +1,40 @@
+(LIR
+  (Declaration W
+    %t1 = load_input "x" :: array
+    loop x id=L1 in %t1 as el=%x_el_2, idx=%x_i_3
+      %t4 = call core.array_size(%x_el_2) :: integer
+      yield %t4
+    end_loop
+  )
+  (Declaration box
+    %t5 = load_input "x" :: array
+    loop x id=L2 in %t5 as el=%x_el_6, idx=%x_i_7
+      %t29 = call core.array_size(%x_el_6) :: integer
+      %t11 = call core.mul(%x_i_7, %t29) :: integer
+      loop y id=L3 in %x_el_6 as el=%y_el_8, idx=%y_i_9
+        %t12 = call core.add(%t11, %y_i_9) :: integer
+        yield %t12
+      end_loop
+    end_loop
+  )
+  (Declaration col_major
+    %t13 = load_input "x" :: array
+    %t19 = call core.array_size(%t13) :: integer
+    loop x id=L4 in %t13 as el=%x_el_14, idx=%x_i_15
+      loop y id=L5 in %x_el_14 as el=%y_el_16, idx=%y_i_17
+        %t20 = call core.mul(%y_i_17, %t19) :: integer
+        %t21 = call core.add(%t20, %x_i_15) :: integer
+        yield %t21
+      end_loop
+    end_loop
+  )
+  (Declaration sum_ij
+    %t22 = load_input "x" :: array
+    loop x id=L6 in %t22 as el=%x_el_23, idx=%x_i_24
+      loop y id=L7 in %x_el_23 as el=%y_el_25, idx=%y_i_26
+        %t27 = call core.add(%x_i_24, %y_i_26) :: integer
+        yield %t27
+      end_loop
+    end_loop
+  )
+)

data/golden/array_index/expected/nast.txt

@@ -0,0 +1,33 @@
+(NAST
+  (VALUE W
+    (Call :"core.array_size"
+    (InputRef [:x, :y])
+  )
+  )
+  (VALUE box
+    (Call :"core.add"
+    (Call :"core.mul"
+      #<struct Kumi::Core::NAST::IndexRef id=83, loc=nil, meta={}>
+      (Ref W)
+    )
+    #<struct Kumi::Core::NAST::IndexRef id=87, loc=nil, meta={}>
+  )
+  )
+  (VALUE col_major
+    (Call :"core.add"
+    (Call :"core.mul"
+      #<struct Kumi::Core::NAST::IndexRef id=91, loc=nil, meta={}>
+      (Call :"core.array_size"
+        (InputRef [:x])
+      )
+    )
+    #<struct Kumi::Core::NAST::IndexRef id=96, loc=nil, meta={}>
+  )
+  )
+  (VALUE sum_ij
+    (Call :"core.add"
+    #<struct Kumi::Core::NAST::IndexRef id=100, loc=nil, meta={}>
+    #<struct Kumi::Core::NAST::IndexRef id=102, loc=nil, meta={}>
+  )
+  )
+)

data/golden/array_index/expected/schema_javascript.mjs

@@ -0,0 +1,46 @@
+export function _box(input) {
+  let out = [];
+  let t5 = input["x"];
+  t5.forEach((x_el_6, x_i_7) => {
+    let out_1 = [];
+    let t29 = x_el_6.length;
+    let t11 = x_i_7 * t29;
+    x_el_6.forEach((y_el_8, y_i_9) => {
+      let t12 = t11 + y_i_9;
+      out_1.push(t12);
+    });
+    out.push(out_1);
+  });
+  return out;
+}
+
+export function _col_major(input) {
+  let out = [];
+  let t13 = input["x"];
+  let t19 = t13.length;
+  t13.forEach((x_el_14, x_i_15) => {
+    let out_1 = [];
+    x_el_14.forEach((y_el_16, y_i_17) => {
+      let t20 = y_i_17 * t19;
+      let t21 = t20 + x_i_15;
+      out_1.push(t21);
+    });
+    out.push(out_1);
+  });
+  return out;
+}
+
+export function _sum_ij(input) {
+  let out = [];
+  let t22 = input["x"];
+  t22.forEach((x_el_23, x_i_24) => {
+    let out_1 = [];
+    x_el_23.forEach((y_el_25, y_i_26) => {
+      let t27 = x_i_24 + y_i_26;
+      out_1.push(t27);
+    });
+    out.push(out_1);
+  });
+  return out;
+}
+