add control cmd

PKUHPC · Dec 25, 2024 · ed5e26d · ed5e26d
1 parent 053cbd0
commit ed5e26d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/plugin/energy/energy.go b/plugin/energy/energy.go
@@ -131,7 +131,7 @@ func (p EnergyPlugin) CreateCgroupHook(ctx *api.PluginContext) {
 	log.Infof("CreateCgroupHook received for cgroup: %s", req.Cgroup)
 
 	boundGPUs := getBoundGPUs(req.Resource, globalMonitor.config.Monitor.GPUType)
-	globalMonitor.monitor.UpdateTaskCount(true)
+	globalMonitor.monitor.NodeMonitor.UpdateTaskCount(true)
 	globalMonitor.monitor.TaskMonitor.Start(req.TaskId, req.Cgroup, boundGPUs)
 }
 
@@ -143,7 +143,7 @@ func (p EnergyPlugin) DestroyCgroupHook(ctx *api.PluginContext) {
 	}
 
 	log.Infof("DestroyCgroupHook received for cgroup: %s", req.Cgroup)
-	globalMonitor.monitor.UpdateTaskCount(false)
+	globalMonitor.monitor.NodeMonitor.UpdateTaskCount(false)
 	globalMonitor.monitor.TaskMonitor.Stop(req.TaskId)
 }
 

diff --git a/plugin/energy/pkg/monitor/monitor.go b/plugin/energy/pkg/monitor/monitor.go
@@ -68,7 +68,3 @@ func (sm *Monitor) Close() {
 	sm.NodeMonitor.Close()
 	sm.TaskMonitor.Close()
 }
-
-func (sm *Monitor) UpdateTaskCount(increment bool) {
-	sm.NodeMonitor.UpdateTaskCount(increment)
-}