From 9bcf6952f7ded4c866aa068acd164eb55990d5b0 Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Sat, 15 Jun 2024 11:43:01 +0200 Subject: [PATCH] Fix flake8 --- ggml_vk_generate_shaders.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ggml_vk_generate_shaders.py b/ggml_vk_generate_shaders.py index ea0c4937a..38914eedb 100644 --- a/ggml_vk_generate_shaders.py +++ b/ggml_vk_generate_shaders.py @@ -4,8 +4,7 @@ import logging import argparse import asyncio import os -import sys -from tempfile import gettempdir, NamedTemporaryFile +from tempfile import gettempdir logger = logging.getLogger("ggml-vk-generate-shaders") @@ -111,7 +110,7 @@ async def main(): # mul mat vec data_a_key = f"DATA_A_{tname.upper()}" - shader = f"mul_mat_vec_{tname}.comp" if tname.endswith("_k") else f"mul_mat_vec.comp" + shader = f"mul_mat_vec_{tname}.comp" if tname.endswith("_k") else "mul_mat_vec.comp" tasks.append(string_to_spv(f"mul_mat_vec_{tname}_f32_f32", shader, base_dict | {data_a_key: "1", "B_TYPE": "float", "D_TYPE": "float"})) tasks.append(string_to_spv(f"mul_mat_vec_{tname}_f16_f32", shader, base_dict | {data_a_key: "1", "B_TYPE": "float16_t", "D_TYPE": "float"}))