put void branch in template

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
2026-02-01 00:27:49 +08:00
parent 875145b333
commit 9a237d40e2
2 changed files with 3 additions and 9 deletions
+1 -8
View File
@@ -56,11 +56,7 @@ class TVMLauncher(object):
shared_memory,
global_scratch,
profile_scratch,
*(
arg
for arg, type in zip(args, self.signature)
if type != "constexpr"
),
*args,
)
)
else:
@@ -164,10 +160,7 @@ class TVMLauncher(object):
)
template: jinja2.Template = env.get_template("launch.c.j2")
signature: List[int] = list(
filter(
lambda t: t != "void",
map(lambda t: type_to_ctype(string_to_type(t)), self.signature),
)
)
html: str = template.render(signature=signature)
return html