Add cli command: list-support-ops

Author: dboyliao

utensor_cgen/backend/base.py

@@ -118,3 +118,7 @@ def from_config(cls, config, *args, **kwargs):
   def from_file(cls, file_or_path, *args, **kwargs):
     config = parse_toml(file_or_path)[cls.TARGET][cls.COMPONENT][cls.PART]
     return cls(config, *args, **kwargs)
+  
+  @property
+  def support_ops(self):
+    return []

utensor_cgen/backend/utensor/_backend_impl.py

@@ -37,6 +37,7 @@ def __init__(
       graph_op_lower = graph_op_lower or uTensorRearchGraphLower(config=config[uTensorRearchGraphLower.PART].to_dict())
       graph_alloc_lower = TensorAllocationPlanner(config=config[TensorAllocationPlanner.PART].to_dict())
     graph_transformer = graph_transformer or PipelineTransformer(config=config[PipelineTransformer.PART].to_dict())
+    self._legacy_api = config['legacy-api']
     self._graph_op_lower = graph_op_lower
     self._graph_transformer = graph_transformer
     self._graph_alloc_lower = graph_alloc_lower
@@ -73,3 +74,11 @@ def __call__(self, ugraph):
   def from_file(cls, path_or_file):
     config = parse_toml(path_or_file)
     return cls(config=config)
+
+  @property
+  def support_ops(self):
+    if self._legacy_api:
+      from .code_generator.legacy._operators import OperatorFactory
+    else:
+      from .code_generator.rearch._operators import OperatorFactory
+    return OperatorFactory.support_op_types()

utensor_cgen/backend/utensor/code_generator/rearch/_operators/_base.py

@@ -23,7 +23,7 @@ def get_opertor(cls, op_info):
     op_cls = cls._operators.get((namespaces, op_type))
     if op_cls is None:
       raise OpNotSupportedError(
-        "{}::{} not supported in utensor_cgen".format("::".join(namespaces), op_type)
+        "{} not supported in utensor_cgen".format("::".join(list(namespaces) + [op_type]))
       )
     return op_cls(op_info)
 
@@ -37,8 +37,11 @@ def register(cls, op_cls):
   @classmethod
   def support_op_types(cls):
     """Return the set of all supported ops
-  """
-    return set(cls._operators.keys())
+    """
+    return set([
+      "::".join(list(namespaces) + [op_type])
+      for namespaces, op_type in cls._operators.keys()
+    ])
 
   @classmethod
   def is_supported(cls, op_type):

utensor_cgen/backend/utensor/code_generator/rearch/_operators/_impls.py

@@ -329,6 +329,37 @@ class _CommonParams(_Operator):
   }
   _ACTIVATION_STR_PATTERN = re.compile(r'^(\d+) \(\w+\)$')
 
+@OperatorFactory.register
+class _Conv2dOperator(_CommonParams):
+  op_type = 'Conv2dOperator'
+
+  @classmethod
+  @must_return_type(Hashable)
+  def get_constructor_parameters(cls, op_info):
+    padding = cls._PADDING_MAP[op_info.op_attr['Padding']]
+    stride_width = op_info.op_attr['StrideW']
+    stride_hight = op_info.op_attr['StrideH']
+    return (
+      _c_arr_str([1, stride_hight, stride_width, 1]),
+      padding,
+    )
+
+  def get_declare_snippet(self, op_var_name, tensor_var_map):
+    return DeclareOpSnippet(
+      op=self,
+      templ_dtypes=[self.out_dtypes[0]],
+      op_var_name=op_var_name,
+      nested_namespaces=self.namespaces,
+    )
+
+  def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
+    return Conv2dOpEvalSnippet(
+      op_info=op_info,
+      templ_dtypes=[self.out_dtypes[0]],
+      op_name=op_var_name,
+      tensor_var_map=tensor_var_map,
+      nested_namespaces=self.namespaces,
+    )
 
 @OperatorFactory.register
 class _QuantDWSConvOperator(_CommonParams):

utensor_cgen/backend/utensor/snippets/rearch/_snippets.py

@@ -10,6 +10,7 @@
   "DeclareRomTensorSnippet",
   "DeclareRamTensorSnippet",
   "DeclareOpSnippet",
+  "Conv2dOpEvalSnippet",
   "DepthwiseSeperateConvOpEvalSnippet",
   "QuantDepthwiseSeperateConvOpEvalSnippet",
   "AddOpEvalSnippet",
@@ -133,6 +134,11 @@ def __init__(self, op_info, templ_dtypes, op_name, tensor_var_map, nested_namesp
     self.template_vars['output_map'] = output_map
 
 
+class Conv2dOpEvalSnippet(OpEvalSnippet):
+  __inputs__ = ["in", "filter"]
+  __outputs__ = ["out"]
+
+
 class DepthwiseSeperateConvOpEvalSnippet(OpEvalSnippet):
   __inputs__ = ["in", "depthwise_filter", "pointwise_filter"]
   __outputs__ = ["out"]

utensor_cgen/cli/backend.py

@@ -1,3 +1,6 @@
+import os
+from pprint import pformat
+
 import click
 
 from utensor_cgen import __version__
@@ -38,6 +41,22 @@ def list_trans_methods(verbose):
     )
   return 0
 
+@cli.command(name='list-support-ops', help='list all supported op in the backend')
+@click.help_option('-h', '--help')
+@click.option('--target', default='utensor')
+@click.option('--config', default='utensor_cli.toml')
+def list_support_ops(target, config):
+  from utensor_cgen.backend.api import BackendManager
+  if os.path.exists(config):
+    backend = BackendManager.get_backend(target).from_file(config)
+  else:
+    backend = BackendManager.get_backend(target)({})
+  click.secho(
+    pformat(backend.support_ops),
+    fg='white',
+    bold=True
+  )
+
 @cli.command(name='generate-config', help='generate config toml file')
 @click.help_option('-h', '--help')
 @click.option('--target', required=True, help='target framework/platform')