chn/nixos
1
0
Fork 0
mirror of https://github.com/CHN-beta/nixos.git synced 2026-01-12 00:40:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

devices.pc: fix ollama rocm gpu support

Browse Source
This commit is contained in:
陈浩南
2025-12-05 21:58:47 +08:00
parent faaff1cc61
commit af3750fa8b
3 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
flake/lib/buildNixpkgsConfig/default.nix
View File

@@ -73,6 +73,7 @@ in platformConfig //
google-chrome = prev.google-chrome.override (prev:
{ commandLineArgs = prev.commandLineArgs or "" + " --disable-features=GlobalShortcutsPortal"; });
xray = prev.xray.overrideAttrs (prev: { patches = prev.patches or [] ++ [ ./xray.patch ]; });
ollama = prev.ollama.overrideAttrs (prev: { patches = prev.patches or [] ++ [ ./ollama.patch ]; });
}
// (
let

20
flake/lib/buildNixpkgsConfig/ollama.patch Normal file
View File

@@ -0,0 +1,20 @@
--- a/discover/runner.go
+++ b/discover/runner.go
@@ -301,7 +301,7 @@ func GPUDevices(ctx context.Context, runners []ml.FilteredRunnerDiscovery) []ml.
// Typical refresh on existing runner is ~500ms but allow longer if the system
// is under stress before giving up and using stale data.
- ctx, cancel := context.WithTimeout(ctx, 3*time.Second)
+ ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
start := time.Now()
updatedDevices := runner.GetDeviceInfos(ctx)
@@ -325,7 +325,7 @@ func GPUDevices(ctx context.Context, runners []ml.FilteredRunnerDiscovery) []ml.
// Bootstrapping may take longer in some cases (AMD windows), but we
// would rather use stale free data to get the model running sooner
- ctx, cancel := context.WithTimeout(ctx, 3*time.Second)
+ ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
// Apply any dev filters to avoid re-discovering unsupported devices, and get IDs correct