Skip to content

Commit

Permalink
Fix flake8
Browse files Browse the repository at this point in the history
0cc4m committed Jun 15, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 00aaaab commit 9bcf695
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions ggml_vk_generate_shaders.py
Original file line number Diff line number Diff line change
@@ -4,8 +4,7 @@
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"}))

0 comments on commit 9bcf695

Please sign in to comment.