ihateninjas/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama synced 2026-04-23 08:45:14 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

gemma4: Disable FA on older GPUs where it doesn't work (#15403)

Browse source 
CUDA older than 7.5 lack the support to enable flash attention for the model.
This commit is contained in:
Daniel Hiltgen 2026-04-07 14:54:25 -07:00 committed by GitHub
parent 780556c4d0
commit dde09129d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
llm/server.go
View file

@ -210,6 +210,18 @@ func NewLlamaServer(systemInfo ml.SystemInfo, gpus []ml.DeviceInfo, modelPath st
fa = false fa = false
} }
// Gemma 4's 512-dim attention heads require MMA FA kernels (Turing+, compute >= 7.5).
// Older CUDA GPUs only have tile/vec FA kernels which abort on dk512 non-GQA attention.
if fa && f.KV().Architecture() == "gemma4" {
for _, gpu := range gpus {
if gpu.Library == "CUDA" && (gpu.ComputeMajor < 7 || (gpu.ComputeMajor == 7 && gpu.ComputeMinor < 5)) {
slog.Debug("disabling flash attention for gemma4 on pre-Turing GPU", "compute", fmt.Sprintf("%d.%d", gpu.ComputeMajor, gpu.ComputeMinor))
fa = false
break
}
}
}
kvct := strings.ToLower(envconfig.KvCacheType()) kvct := strings.ToLower(envconfig.KvCacheType())
if tok == nil { if tok == nil {