UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-25 21:39:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'CNT-3963/deduplicate-wsl-driverstore-paths' into 'main'

Browse Source 
Deduplicate WSL driverstore paths

See merge request nvidia/container-toolkit/container-toolkit!304
This commit is contained in:
Evan Lezar 2023-02-27 13:27:31 +00:00
commit 7986ff9cee
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/dxcore/api.go
View File

@ -49,9 +49,15 @@ func Shutdown() error {
// GetDriverStorePaths returns the list of driver store paths // GetDriverStorePaths returns the list of driver store paths
func GetDriverStorePaths() []string { func GetDriverStorePaths() []string {
var paths []string var paths []string
selected := make(map[string]bool)
for i := 0; i < dxcore.getAdapterCount(); i++ { for i := 0; i < dxcore.getAdapterCount(); i++ {
adapter := dxcore.getAdapter(i) path := dxcore.getAdapter(i).getDriverStorePath()
paths = append(paths, adapter.getDriverStorePath()) if selected[path] {
continue
}
selected[path] = true
paths = append(paths, path)
} }
return paths return paths