diff --git a/gpu/amd_hip_windows.go b/gpu/amd_hip_windows.go index 99d37c17..2cea2824 100644 --- a/gpu/amd_hip_windows.go +++ b/gpu/amd_hip_windows.go @@ -34,10 +34,10 @@ type HipLib struct { } func NewHipLib() (*HipLib, error) { - // At runtime we depend on v6, so discover GPUs with the same library for a consistent set of GPUs/ this repo will consist with v5.7 - h, err := windows.LoadLibrary("amdhip64.dll") + // At runtime we depend on v6, so discover GPUs with the same library for a consistent set of GPUs + h, err := windows.LoadLibrary("amdhip64_6.dll") if err != nil { - return nil, fmt.Errorf("unable to load amdhip64.dll, please make sure to upgrade to the latest amd driver: %w", err) + return nil, fmt.Errorf("unable to load amdhip64_6.dll, please make sure to upgrade to the latest amd driver: %w", err) } hl := &HipLib{} hl.dll = h diff --git a/gpu/amd_windows.go b/gpu/amd_windows.go index 95f32cb9..f2896e21 100644 --- a/gpu/amd_windows.go +++ b/gpu/amd_windows.go @@ -23,7 +23,7 @@ const ( var ( // Used to validate if the given ROCm lib is usable ROCmLibGlobs = []string{"hipblas.dll", "rocblas"} // This is not sufficient to discern v5 vs v6 - RocmStandardLocations = []string{"C:\\Program Files\\AMD\\ROCm\\5.7\\bin"} // TODO glob? + RocmStandardLocations = []string{"C:\\Program Files\\AMD\\ROCm\\6.1\\bin"} // TODO glob? ) func AMDGetGPUInfo() []RocmGPUInfo {