Add simplified SingleOp test generation helper system

uTensor · mbartling · May 15, 2020 · May 21, 2020 · May 23, 2020 · May 23, 2020
commit 45473d76847372f61221cf4815bbb17b11f093dd
diff --git a/.gitignore b/.gitignore
@@ -39,6 +39,7 @@ sd-driver
 #Project generated test files
 python/test_scripts/testData
 python/.venv
+__pycache__
 
 #Vim stuff
 *.swp

diff --git a/python/test_scripts/__init__.py b/python/test_scripts/__init__.py
diff --git a/python/test_scripts/gen_softmax.py b/python/test_scripts/gen_softmax.py
@@ -0,0 +1,24 @@
+from jinja_env import env2, Operator, Tensor, SingleOpTest
+import tensorflow as tf
+
+test_group = "Softmax"
+
+def gen_test(test_number, scale = 1.0):
+    test_name = "random_gen_scale_%d__%d" % ( int(scale), test_number)
+    in1 = tf.constant(tf.random.uniform([1,10])*scale).numpy()
+    out_1 = tf.nn.softmax(in1).numpy()
+
+    in_ref_name = "s_ref_in_%d" % test_number
+    out_ref_name = "s_ref_out_%d" % test_number
+    in_t = Tensor("in", in1, ref_name=in_ref_name)        
+    out_ref = Tensor("out_ref", out_1, ref_name=out_ref_name) # Store the reference out values
+    out_t = Tensor("out", out_1)                 
+    #conv_param_str = "{%s}, %s" % (str(strides).lstrip('[').rstrip(']'), padding)
+    #convOp = Operator("Conv2dOperator", "op_0", dtypes=["float"], param_str=conv_param_str)
+    op = Operator("SoftmaxOperator", "softmaxOp", dtypes=["float"])
+    op.set_inputs({"in": in_t}).set_outputs({"out": out_t})
+
+    test = SingleOpTest(test_group, test_name, op)
+    test.add_tensor_comparison(out_t, out_ref)
+    test_rendered, const_snippets = test.render()
+    print(test_rendered)
diff --git a/python/test_scripts/jinja_env/__init__.py b/python/test_scripts/jinja_env/__init__.py
@@ -1,23 +1,177 @@
 import jinja2
 from pathlib import Path
+import numpy as np
 
 _template_dir = Path(__file__).parent / "templates"
+_template2_dir = Path(__file__).parent / "templates_v2"
 
 env = jinja2.Environment(
     loader=jinja2.FileSystemLoader(_template_dir), trim_blocks=True, lstrip_blocks=True
-)
+    )
 env.globals.update(
     zip=zip,
     len=len,
     TENSOR_TYPE_MAP={
-        "int8_t": "i8",
-        "uint8_t": "u8",
-        "int16_t": "i16",
-        "uint16_t": "u16",
-        "int32_t": "i32",
-        "uint32_t": "u32",
-        "float": "flt",
-    },
-)
+      "int8_t": "i8",
+      "uint8_t": "u8",
+      "int16_t": "i16",
+      "uint16_t": "u16",
+      "int32_t": "i32",
+      "uint32_t": "u32",
+      "float": "flt",
+      },
+    )
 
 del _template_dir
+
+env2 = jinja2.Environment(
+    loader=jinja2.FileSystemLoader(_template2_dir), trim_blocks=True, lstrip_blocks=True
+    )
+env2.globals.update(
+    zip=zip,
+    len=len,
+    TENSOR_TYPE_MAP={
+      "int8_t": "i8",
+      "uint8_t": "u8",
+      "int16_t": "i16",
+      "uint16_t": "u16",
+      "int32_t": "i32",
+      "uint32_t": "u32",
+      "float": "flt",
+      },
+    NUMPY_2_CMAP={
+      np.int8: "int8_t",
+      np.uint8: "uint8_t",
+      np.int16: "int16_t",
+      np.uint16: "uint16_t",
+      np.int32: "int32_t",
+      np.uint32: "uint32_t",
+      np.float: "float",
+      np.dtype('float32'): "float",
+      },
+    )
+
+TENSOR_TYPE_MAP={
+    "int8_t": "i8",
+    "uint8_t": "u8",
+    "int16_t": "i16",
+    "uint16_t": "u16",
+    "int32_t": "i32",
+    "uint32_t": "u32",
+    "float": "flt",
+    }
+NUMPY_2_CMAP={
+    np.int8: "int8_t",
+    np.uint8: "uint8_t",
+    np.int16: "int16_t",
+    np.uint16: "uint16_t",
+    np.int32: "int32_t",
+    np.uint32: "uint32_t",
+    np.float: "float",
+    np.dtype('float32'): "float",
+    }
+
+class Tensor:
+  def __init__(self, name, np_array, ref_name=None, quantize_params=[]):
+    self.name = name
+    self.np_array = np_array
+    self.ref_name = ref_name
+    self.quantize_params = quantize_params
+
+  @property
+  def shape(self):
+    return self.np_array.shape
+
+  @property
+  def dtype(self):
+    return NUMPY_2_CMAP[self.np_array.dtype]
+
+  @property
+  def utype(self):
+    return TENSOR_TYPE_MAP[self.dtype]
+
+  def flatten(self):
+    return self.np_array.flatten()
+
+  def render_constant(self):
+    if self.ref_name:
+      return env2.get_template('def_constant.hpp').render(tensor=self)
+    else:
+      return ""
+  def render_declaration(self):
+    if self.ref_name:
+      return env2.get_template('declare_rom_tensor.cpp').render(tensor=self)
+    else:
+      return env2.get_template('declare_ram_tensor.cpp').render(tensor=self)
+
+
+class Operator:
+  def __init__(self, op_type, name, dtypes=[], param_str=None):
+    self.op_type = op_type
+    self.name = name
+    self.dtypes = dtypes
+    self.param_str = param_str
+    self.array_template = env2.get_template('array_template.cpp')
+    self.input_map = {}
+    self.output_map = {}
+    self.type_signature = env2.get_template('op_type_signature.cpp').render(op=self)
+
+  def set_inputs(self, input_map):
+    self.input_map = input_map
+    return self
+
+  def set_outputs(self, output_map):
+    self.output_map = output_map
+    return self
+
+  def render_declaration(self):
+    return env2.get_template('declare_operator.cpp').render(op=self)
+
+  def render_eval(self):
+    return env2.get_template('eval_operator.cpp').render(op=self)
+
+class SingleOpTest:
+  def __init__(self, test_group, test_name, target_op):
+    self.test_group = test_group
+    self.test_name = test_name
+    self.out_size = 0
+    for out_tensor in target_op.output_map:
+      self.out_size += len(target_op.output_map[out_tensor].flatten())
+    self.target_op = target_op
+    self.compare_tensors = []
+    self.tensor_set = set()
+    for tensor in target_op.input_map:
+      self.tensor_set.add(target_op.input_map[tensor])
+    for tensor in target_op.output_map:
+      self.tensor_set.add(target_op.output_map[tensor])
+
+  def add_tensor_comparison(self, a, b):
+    self.compare_tensors.append((a,b))
+    self.tensor_set.add(a)
+    self.tensor_set.add(b)
+
+  def render(self):
+    const_snippets = []
+    tensor_decls = []
+    for tensor in self.tensor_set:
+      const_snippets.append(tensor.render_constant())
+      tensor_decls.append(tensor.render_declaration())
+    op_decl = self.target_op.render_declaration()
+    op_eval = self.target_op.render_eval()
+
+    compare_snippets = []
+    for a, b in self.compare_tensors:
+      compare_snippets.append(env2.get_template('compare_outputs.cpp').render(a=a, b=b))
+
+    TestTemplate = env2.get_template('test_container.cpp')
+    test_rendered = TestTemplate.render(test_group= self.test_group, 
+                                        test_name = self.test_name,
+                                        out_size  = self.out_size,
+                                        tensor_declarations = tensor_decls,
+                                        op_decl = op_decl,
+                                        op_eval = op_eval,
+                                        compare_snippets=compare_snippets)
+    return (test_rendered, const_snippets)
+
+
+del _template2_dir
diff --git a/python/test_scripts/jinja_env/templates_v2/array_template.cpp b/python/test_scripts/jinja_env/templates_v2/array_template.cpp
@@ -0,0 +1 @@
+{% for x in arr %}{{ x }}{{ "," if not loop.last }}{% endfor %}
diff --git a/python/test_scripts/jinja_env/templates_v2/compare_outputs.cpp b/python/test_scripts/jinja_env/templates_v2/compare_outputs.cpp
@@ -0,0 +1,3 @@
+for(int i = 0; i < {{ len(b.flatten()) }}; i++) {
+  EXPECT_NEAR(static_cast<{{ a.dtype }}>( {{ a.name }}(i) ), static_cast<{{ b.dtype }}>( {{ b.name }}(i) ), 0.0001);
+}
diff --git a/python/test_scripts/jinja_env/templates_v2/const_container.hpp b/python/test_scripts/jinja_env/templates_v2/const_container.hpp
@@ -0,0 +1,6 @@
+#ifndef {{ constants_header | replace(".", "_") }} 
+#define {{ constants_header | replace(".", "_") }} 
+{% for constant_snippet in constants %}
+{{ constant_snippet }}
+{% endfor %}
+#endif
diff --git a/python/test_scripts/jinja_env/templates_v2/declare_operator.cpp b/python/test_scripts/jinja_env/templates_v2/declare_operator.cpp
@@ -0,0 +1 @@
+{{ op.type_signature }} {{ op.name }}{% if op.param_str %}({{ op.param_str }}){% endif %};
diff --git a/python/test_scripts/jinja_env/templates_v2/declare_ram_tensor.cpp b/python/test_scripts/jinja_env/templates_v2/declare_ram_tensor.cpp
@@ -0,0 +1,4 @@
+Tensor {{tensor.name}} = new RamTensor({ {%for s in tensor.shape%}{{ s }}{{"," if not loop.last}}{%endfor%} }, {{ tensor.utype }});
+{%if tensor.quantize_params%}
+  {{tensor.name}}->set_quantization_params(PerTensorQuantizationParams({{tensor.quantize_params[1]}}, {{tensor.quantize_params[0]}}));
+{%endif%}
diff --git a/python/test_scripts/jinja_env/templates_v2/declare_rom_tensor.cpp b/python/test_scripts/jinja_env/templates_v2/declare_rom_tensor.cpp
@@ -0,0 +1,4 @@
+Tensor {{ tensor.name }} = new RomTensor({ {% for s in tensor.shape %}{{ s }}{{"," if not loop.last}}{% endfor %} }, {{ tensor.utype }}, {{ tensor.ref_name }});
+{%if tensor.quantize_params%}
+  {{tensor.name}}->set_quantization_params(PerTensorQuantizationParams({{tensor.quantize_params[1]}}, {{tensor.quantize_params[0]}}));
+{%endif%}
diff --git a/python/test_scripts/jinja_env/templates_v2/def_constant.hpp b/python/test_scripts/jinja_env/templates_v2/def_constant.hpp
@@ -0,0 +1,3 @@
+static const {{ tensor.dtype }} {{ tensor.ref_name }}[{{ len(tensor.flatten()) }}] = { 
+  {% for x in tensor.flatten() %} {{ x }}{{ "," if not loop.last }}{{ "\n" if not loop.first and loop.index % 10 == 0}} {% endfor %} 
+};
diff --git a/python/test_scripts/jinja_env/templates_v2/eval_operator.cpp b/python/test_scripts/jinja_env/templates_v2/eval_operator.cpp
@@ -0,0 +1,10 @@
+{{op.name}}
+  .set_inputs({ 
+    {% for x in op.input_map %}
+    { {{op.type_signature}}::{{x}}, {{op.input_map[x].name}} }{{"," if not loop.last}}
+  {% endfor %}
+  }).set_outputs({ 
+    {% for x in op.output_map %}
+    { {{op.type_signature}}::{{x}}, {{op.output_map[x].name}} }{{"," if not loop.last}}
+  {% endfor %}
+  }).eval();
diff --git a/python/test_scripts/jinja_env/templates_v2/op_type_signature.cpp b/python/test_scripts/jinja_env/templates_v2/op_type_signature.cpp
@@ -0,0 +1 @@
+{{ op.op_type }}{% if op.dtypes %}<{{ op.array_template.render(arr=op.dtypes) }}>{% endif %}
diff --git a/python/test_scripts/jinja_env/templates_v2/test_container.cpp b/python/test_scripts/jinja_env/templates_v2/test_container.cpp
@@ -0,0 +1,20 @@
+/***************************************
+ * Generated Test
+ ***************************************/
+
+TEST({{ test_group }}, {{ test_name }}) {
+  localCircularArenaAllocator<1024> meta_allocator;
+  localCircularArenaAllocator<{{ out_size }}*2*sizeof(float), uint32_t> ram_allocator;
+  Context::get_default_context()->set_metadata_allocator(&meta_allocator);
+  Context::get_default_context()->set_ram_data_allocator(&ram_allocator);
+
+  {% for tensor_decl in tensor_declarations %}
+  {{ tensor_decl }}{% endfor %}
+
+  {{ op_decl }}
+  {{ op_eval }}
+
+  {% for compare_snippet in compare_snippets %}
+  {{ compare_snippet }}
+  {% endfor %}
+}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{% for x in arr %}{{ x }}{{ "," if not loop.last }}{% endfor %}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{{ op.type_signature }} {{ op.name }}{% if op.param_str %}({{ op.param_str }}){% endif %};
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{{ op.op_type }}{% if op.dtypes %}<{{ op.array_template.render(arr=op.dtypes) }}>{% endif %}