diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/common/primitive_db_gen.py b/inference-engine/thirdparty/clDNN/kernel_selector/core/common/primitive_db_gen.py index 0901ce27b93775..2adee67fc31b76 100644 --- a/inference-engine/thirdparty/clDNN/kernel_selector/core/common/primitive_db_gen.py +++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/common/primitive_db_gen.py @@ -9,6 +9,7 @@ import argparse import glob import ntpath +import re class OpenCL2CHeaders(object): @@ -64,6 +65,27 @@ def cl_file_to_str(self, filename): max_characters = 16350 characters = 1 # Newline character above + comment_regexp = re.compile(r'(^)?[^\S\n]*/(?:\*(.*?)\*/[^\S\n]*|/[^\n]*)($)?', re.DOTALL | re.MULTILINE) + + def comment_replacer(match): + begin, mid, end = match.group(1,2,3) + if mid is None: + return '' + elif begin is not None or end is not None: + return '' + elif '\n' in mid: + return '\n' + else: + return ' ' + + return + + # Remove comments + content = comment_regexp.sub(comment_replacer, content) + # Remove empty lines + content = os.linesep.join([s for s in content.splitlines() if s]) + # Remove multiple spaces + content = re.sub(' +', ' ', content) for i, line in enumerate(content.split('\n')): if (i + 1) % max_lines == 0 or characters + len(line) + 1 > max_characters: res += ')__krnl"\n + R"__krnl('