From 8951ddf780e02616cdb2ec54a02bc354e8507c0f Mon Sep 17 00:00:00 2001 From: iaz1607 <iaz1607@yandex-team.com> Date: Thu, 30 Nov 2023 12:16:39 +0300 Subject: `build/scripts` ya style --py --- build/scripts/compile_cuda.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'build/scripts/compile_cuda.py') diff --git a/build/scripts/compile_cuda.py b/build/scripts/compile_cuda.py index f8e1fa2b6d..eadb4519d2 100644 --- a/build/scripts/compile_cuda.py +++ b/build/scripts/compile_cuda.py @@ -27,8 +27,8 @@ def main(): if sys.argv[1] == '--mtime': mtime0 = sys.argv[2] cmd = 3 - command = sys.argv[cmd: spl] - cflags = sys.argv[spl + 1:] + command = sys.argv[cmd:spl] + cflags = sys.argv[spl + 1 :] dump_args = False if '--y_dump_args' in command: @@ -85,16 +85,20 @@ def main(): for flag in cflags: if all(not flag.startswith(skip_prefix) for skip_prefix in skip_prefix_list): if flag.startswith('-fopenmp-version='): - new_cflags.append('-fopenmp-version=45') # Clang 11 only supports OpenMP 4.5, but the default is 5.0, so we need to forcefully redefine it. + new_cflags.append( + '-fopenmp-version=45' + ) # Clang 11 only supports OpenMP 4.5, but the default is 5.0, so we need to forcefully redefine it. else: new_cflags.append(flag) cflags = new_cflags if not is_clang(command): + def good(arg): if arg.startswith('--target='): return False return True + cflags = filter(good, cflags) cpp_args = [] @@ -108,7 +112,6 @@ def main(): cflags_queue = collections.deque(cflags) while cflags_queue: - arg = cflags_queue.popleft() if arg == '-mllvm': compiler_args.append(arg) -- cgit v1.2.3