Merge pull request volcano-sh#3054 from archlitchi/nodelockrm

Add nodelockEnable parameters default to false
gj409237405 · Aug 14, 2023 · 2cd8ae5 · 2cd8ae5
2 parents 04a3b1b + 1b4f137
commit 2cd8ae5
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)