merge fixes
This commit is contained in:
parent
294b179688
commit
c4d8c75e54
|
|
@ -3,7 +3,6 @@ package discover
|
|||
// Runner based GPU discovery
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
|
|
|||
|
|
@ -174,7 +174,7 @@ func (l GpuInfoList) FlashAttentionSupported() bool {
|
|||
supportsFA := gpu.Library == "cpu" ||
|
||||
gpu.Name == "Metal" || gpu.Library == "Metal" ||
|
||||
(gpu.Library == "CUDA" && gpu.DriverMajor >= 7) ||
|
||||
gpu.Library == "ROCm"
|
||||
gpu.Library == "ROCm" ||
|
||||
gpu.Library == "Vulkan"
|
||||
|
||||
if !supportsFA {
|
||||
|
|
|
|||
|
|
@ -18,7 +18,6 @@ import (
|
|||
"log/slog"
|
||||
"maps"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"slices"
|
||||
"strconv"
|
||||
|
|
|
|||
Loading…
Reference in New Issue