Fix bug: MaxPool op is not listed in the supported ops

uTensor · Dec 16, 2020 · fa8cdb4 · fa8cdb4
1 parent fe8abc4
commit fa8cdb4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/utensor_cgen/backend/utensor/code_generator/rearch/_operators/_impls.py b/utensor_cgen/backend/utensor/code_generator/rearch/_operators/_impls.py
@@ -543,7 +543,7 @@ def get_construct_snippet(self, op_var_name):
 
 
 @OperatorFactory.register
-class _MaxPoolOperator(_GenericMaxOperator, _PoolingOperatorMixin):
+class _MaxPoolOperator(_GenericMaxPoolOperator, _PoolingOperatorMixin):
   namespaces = ('ReferenceOperators',)
 
   def get_declare_snippet(self, op_var_name, with_const_params=True):

diff --git a/utensor_cgen/cli/backend.py b/utensor_cgen/cli/backend.py
@@ -51,6 +51,11 @@ def list_support_ops(target, config):
     backend = BackendManager.get_backend(target).from_file(config)
   else:
     backend = BackendManager.get_backend(target)({})
+  click.secho(
+    f"Supported ops in {target!r}:",
+    fg="green",
+    bold=True,
+  )
   click.secho(
     pformat(backend.support_ops),
     fg='white',