thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Added initial implementation of shaders as hpp files

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-23 10:56:14 +01:00
parent 2561ace8b7
commit f3b28e6b08
9 changed files with 30 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

3
scripts/convert_shaders.py
View file

@ -80,11 +80,12 @@ def run_cli(
logger.debug(f"Header path provided. Converting bin files to hpp.")
logger.debug(f"Output header path: {shader_path}")
for file in spirv_files:
logger.debug(f"Converting to hpp: {file}")
header_data = str(sh.xxd("-i", file))
file_name = file.split("/")[-1]
header_file = file_name.replace(".comp.spv", ".hpp")
logger.debug(f"Converting to hpp: {file_name}")
with open(os.path.join(header_path, header_file), "w+") as fstream:
fstream.write("#pragma once\n\n")
fstream.write("namespace kp {\n")
fstream.write("namespace shader_data {\n")
fstream.write(header_data)