Merge pull request #13022 from zexi/gpu-fuzzy-match

fix(region): gpu fuzzy match model name exactly
This commit is contained in:
Zexi Li
2021-12-24 18:09:40 +08:00
committed by GitHub
+7 -3
View File
@@ -328,8 +328,13 @@ func (manager *SIsolatedDeviceManager) fuzzyMatchModel(fuzzyStr string) *SIsolat
dev := SIsolatedDevice{}
dev.SetModelManager(manager, &dev)
q := manager.Query().Contains("model", fuzzyStr)
err := q.First(&dev)
q := manager.Query().Equals("model", fuzzyStr)
cnt, err := q.CountWithError()
if err != nil || cnt == 0 {
q = manager.Query().Contains("model", fuzzyStr)
}
err = q.First(&dev)
if err == nil {
return &dev
}
@@ -379,7 +384,6 @@ func (manager *SIsolatedDeviceManager) parseDeviceInfo(userCred mcclient.TokenCr
} else {
devConfig.Vendor = matchDev.getVendorId()
}
} else {
devObj, err := manager.FetchById(devId)
if err != nil {