llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Importer] fix has value semantic return type (#2404) 

* fix value semantic return

* address comments

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
pull/2275/merge snapshot-20230829.945
Zhekun(Josh) Zhang 2023-08-28 19:14:09 -07:00 committed by GitHub
parent ca34b9c4fc
commit 5282324c68
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/registry.py
View File

@ -199,8 +199,11 @@ class JitOperator:
parameter_decls = list(map(parameter_decl_builder, self.arguments)) parameter_decls = list(map(parameter_decl_builder, self.arguments))
parameter_decls = list(filter(None, parameter_decls)) parameter_decls = list(filter(None, parameter_decls))
ret_decls = list(map(ret_decl_builder, self.returns)) ret_decls = list(map(ret_decl_builder, self.returns))
ret_decls = list(filter(None, ret_decls))
parameters = ", ".join(parameter_decls) parameters = ", ".join(parameter_decls)
result = ", ".join(ret_decls) result = ", ".join(ret_decls)
if len(ret_decls) == 0:
result = "None"
if len(ret_decls) >= 2: if len(ret_decls) >= 2:
result = f"Tuple[{result}]" result = f"Tuple[{result}]"
@ -286,7 +289,7 @@ class JitOperator:
return "" return ""
def ret_decl_builder(arg: "SIG_ATTR_TYPE") -> str: def ret_decl_builder(arg: "SIG_ATTR_TYPE") -> str:
return "None" return ""
return self._get_function_signature( return self._get_function_signature(
"has_value_semantics", parameter_decl_builder, ret_decl_builder) "has_value_semantics", parameter_decl_builder, ret_decl_builder)