From f6dd7070deba990c560accebf431140c970d4cb6 Mon Sep 17 00:00:00 2001 From: Inforithmics Date: Sun, 10 Aug 2025 21:22:26 +0200 Subject: [PATCH] vk_check_flash_attention 0 means supported --- discover/gpu.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/discover/gpu.go b/discover/gpu.go index a77f8896b..13c607b18 100644 --- a/discover/gpu.go +++ b/discover/gpu.go @@ -423,7 +423,7 @@ func GetGPUInfo() GpuInfoList { continue } - gpuInfo.FlashAttention = (C.vk_check_flash_attention(*vHandles.vulkan, C.int(i)) == 1) // 1 means supported + gpuInfo.FlashAttention = (C.vk_check_flash_attention(*vHandles.vulkan, C.int(i)) == 0) // 0 means supported gpuInfo.TotalMemory = uint64(memInfo.total) gpuInfo.FreeMemory = uint64(memInfo.free) gpuInfo.ID = C.GoString(&memInfo.gpu_id[0])