Filter out empty strings while generting function signature

2023-07-12 23:23:43 +00:00 · 2023-07-12 23:23:43 +00:00 · 91c6454618
parent c23a61f4b6
commit 91c6454618
1 changed files with 1 additions and 0 deletions
--- a/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/registry.py
+++ b/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/registry.py
@ -197,6 +197,7 @@ class JitOperator:
        def_name = "〇".join(mlir_op_name.split("."))
        def_name += f"〡{function_kind}"
        parameter_decls = list(map(parameter_decl_builder, self.arguments))
+        parameter_decls = list(filter(None, parameter_decls))
        ret_decls = list(map(ret_decl_builder, self.returns))
        parameters = ", ".join(parameter_decls)
        result = ", ".join(ret_decls)