Adding support for deinitialization.

Author: isazi

kernel_tuner/utils/directives.py

@@ -380,6 +380,22 @@ def extract_initialization_code(code: str, langs: Code) -> str:
         return ""
 
 
+def extract_deinitialization_code(code: str, langs: Code) -> str:
+    """Extract the deinitialization section from code"""
+    if is_cxx(langs.language):
+        start_string = "#pragma tuner deinitialize"
+        end_string = "#pragma tuner stop"
+    elif is_fortran(langs.language):
+        start_string = "!$tuner deinitialize"
+        end_string = "!$tuner stop"
+
+    init_code = extract_code(start_string, end_string, code, langs)
+    if len(init_code) >= 1:
+        return "\n".join(init_code.values()) + "\n"
+    else:
+        return ""
+
+
 def format_argument_fortran(p_type: str, p_size: int, p_name: str) -> str:
     """Format the argument for Fortran code"""
     argument = ""

test/utils/test_directives.py

@@ -326,6 +326,13 @@ def test_extract_initialization_code():
     assert extract_initialization_code(code_f90, Code(OpenACC(), Fortran())) == "integer :: value\n"
 
 
+def test_extract_deinitialization_code():
+    code_cpp = "#pragma tuner deinitialize\nconst int value = 42;\n#pragma tuner stop\n"
+    code_f90 = "!$tuner deinitialize\ninteger :: value\n!$tuner stop\n"
+    assert extract_deinitialization_code(code_cpp, Code(OpenACC(), Cxx())) == "const int value = 42;\n"
+    assert extract_deinitialization_code(code_f90, Code(OpenACC(), Fortran())) == "integer :: value\n"
+
+
 def test_add_present_openacc():
     acc_cxx = Code(OpenACC(), Cxx())
     acc_f90 = Code(OpenACC(), Fortran())