Merge pull request #3060 from wangyang0616/fix_nodelock

[cherry-pick for release-1.8] fix nodelock issue when using gang-scheduling
volcano-sh · Aug 17, 2023 · aa57168 · aa57168
2 parents f03837a + 89124f0
commit aa57168
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pkg/scheduler/api/devices/nvidia/vgpu/device_info.go b/pkg/scheduler/api/devices/nvidia/vgpu/device_info.go
@@ -204,10 +204,12 @@ func (gs *GPUDevices) Allocate(kubeClient kubernetes.Interface, pod *v1.Pod) err
 			klog.Errorln("DeviceSharing err=", err.Error())
 			return err
 		}
-		nodelock.UseClient(kubeClient)
-		err = nodelock.LockNode(gs.Name, DeviceName)
-		if err != nil {
-			return errors.Errorf("node %s locked for lockname gpushare %s", gs.Name, err.Error())
+		if NodeLockEnable {
+			nodelock.UseClient(kubeClient)
+			err = nodelock.LockNode(gs.Name, DeviceName)
+			if err != nil {
+				return errors.Errorf("node %s locked for lockname gpushare %s", gs.Name, err.Error())
+			}
 		}
 
 		annotations := make(map[string]string)