define coopmat or it will segfault

This commit is contained in:
Concedo 2025-01-09 16:38:21 +08:00
commit 5cce8a5fbc
3 changed files with 31 additions and 3 deletions

View file

@ -34,6 +34,9 @@
#include <vulkan/vulkan_core.h>
#define ASYNCIO_CONCURRENCY 64
#ifndef GGML_VULKAN_COOPMAT_GLSLC_SUPPORT
#define GGML_VULKAN_COOPMAT_GLSLC_SUPPORT
#endif
std::mutex lock;
std::vector<std::pair<std::string, std::string>> shader_fnames;
@ -347,9 +350,11 @@ void process_shaders() {
matmul_shaders(true, matmul_id, false, false, false);
matmul_shaders(true, matmul_id, false, false, true);
#if defined(GGML_VULKAN_COOPMAT_GLSLC_SUPPORT)
// Coopmat, fp32acc and fp16acc
matmul_shaders(true, matmul_id, true, false, false);
matmul_shaders(true, matmul_id, true, false, true);
#endif
#if defined(GGML_VULKAN_COOPMAT2_GLSLC_SUPPORT)
// Coopmat2, fp32acc and fp16acc