llama-cpp-turboquant/python/test/utils.py
2021-09-14 23:09:37 +01:00

8 lines
228 B
Python

import os
def compile_source(source):
open("tmp_kp_shader.comp", "w").write(source)
os.system("glslangValidator -V tmp_kp_shader.comp -o tmp_kp_shader.comp.spv")
return open("tmp_kp_shader.comp.spv", "rb").read()