Jelajahi Sumber

Only enable numa on CPUs (#6484)

The numa flag may be having a performance impact on multi-socket systems with GPU loads
Daniel Hiltgen 8 bulan lalu
induk
melakukan
0f92b19bec
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      llm/server.go

+ 1 - 1
llm/server.go

@@ -258,7 +258,7 @@ func NewLlamaServer(gpus gpu.GpuInfoList, model string, ggml *GGML, adapters, pr
 		params = append(params, "--mlock")
 	}
 
-	if gpu.IsNUMA() {
+	if gpu.IsNUMA() && gpus[0].Library == "cpu" {
 		numaMode := "distribute"
 		if runtime.GOOS == "linux" {
 			if _, err := exec.LookPath("numactl"); err == nil {