Skip to content

Commit ae532ea

Browse files
authored
vulkan: disable spirv-opt for bfloat16 shaders (#15352)
1 parent e5155e6 commit ae532ea

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,8 @@ void string_to_spv_func(const std::string& _name, const std::string& in_fname, c
223223
std::string target_env = (name.find("_cm2") != std::string::npos) ? "--target-env=vulkan1.3" : "--target-env=vulkan1.2";
224224

225225
// disable spirv-opt for coopmat shaders for https://github.com/ggerganov/llama.cpp/issues/10734
226-
std::string opt_level = coopmat ? "" : "-O";
226+
// disable spirv-opt for bf16 shaders for https://github.com/ggml-org/llama.cpp/issues/15344
227+
std::string opt_level = (coopmat || name.find("bf16") != std::string::npos) ? "" : "-O";
227228

228229
#ifdef _WIN32
229230
std::vector<std::string> cmd = {GLSLC, "-fshader-stage=compute", target_env, opt_level, "\"" + in_path + "\"", "-o", "\"" + out_fname + "\""};

0 commit comments

Comments
 (0)