mirror of
https://github.com/sgjzfzzf/triton-tvm-ffi.git
synced 2026-07-01 08:51:56 +08:00
put void branch in template
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user