Skip to content

Commit 083c946

Browse files
committed
Use deinitialization.
1 parent cdd7540 commit 083c946

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

kernel_tuner/utils/directives.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -524,6 +524,7 @@ def generate_directive_function(
524524
langs: Code,
525525
data: dict = None,
526526
initialization: str = "",
527+
deinitialization: str = "",
527528
user_dimensions: dict = None,
528529
) -> str:
529530
"""Generate tunable function for one directive"""
@@ -551,13 +552,17 @@ def generate_directive_function(
551552
else:
552553
code += body
553554
code = end_timing_cxx(code)
555+
if len(deinitialization) > 1:
556+
code += deinitialization + "\n"
554557
code += "\n}"
555558
elif is_fortran(langs.language):
556559
body = wrap_timing(body, langs.language)
557560
if data is not None:
558561
code += wrap_data(body + "\n", langs, data, preprocessor, user_dimensions)
559562
else:
560563
code += body
564+
if len(deinitialization) > 1:
565+
code += deinitialization + "\n"
561566
name = signature.split(" ")[1].split("(")[0]
562567
code += f"\nend function {name}\nend module kt\n"
563568

0 commit comments

Comments
 (0)