浏览代码

Fix missing dep path on windows CPU runners (#6884)

GPUs handled the dependency path properly, but CPU runners didn't which
results in missing vc redist libraries on systems where the user didn't
already have it installed from some other app.
Daniel Hiltgen 7 月之前
父节点
当前提交
6c2eb73a70
共有 1 个文件被更改,包括 7 次插入6 次删除
  1. 7 6
      gpu/gpu.go

+ 7 - 6
gpu/gpu.go

@@ -205,13 +205,16 @@ func GetGPUInfo() GpuInfoList {
 		if err != nil {
 		if err != nil {
 			slog.Warn("error looking up system memory", "error", err)
 			slog.Warn("error looking up system memory", "error", err)
 		}
 		}
+		depPath := LibraryDir()
+
 		cpus = []CPUInfo{
 		cpus = []CPUInfo{
 			{
 			{
 				GpuInfo: GpuInfo{
 				GpuInfo: GpuInfo{
-					memInfo: mem,
-					Library: "cpu",
-					Variant: cpuCapability.String(),
-					ID:      "0",
+					memInfo:        mem,
+					Library:        "cpu",
+					Variant:        cpuCapability.String(),
+					ID:             "0",
+					DependencyPath: depPath,
 				},
 				},
 			},
 			},
 		}
 		}
@@ -224,8 +227,6 @@ func GetGPUInfo() GpuInfoList {
 			return GpuInfoList{cpus[0].GpuInfo}
 			return GpuInfoList{cpus[0].GpuInfo}
 		}
 		}
 
 
-		depPath := LibraryDir()
-
 		// Load ALL libraries
 		// Load ALL libraries
 		cHandles = initCudaHandles()
 		cHandles = initCudaHandles()