Register resourceaware plugin in factory.go; restore priority guard clause

Author: CoolingCube

pkg/scheduler/plugins/factory.go

@@ -45,12 +45,14 @@ import (
 	"github.com/kai-scheduler/KAI-scheduler/pkg/scheduler/plugins/subgrouporder"
 	"github.com/kai-scheduler/KAI-scheduler/pkg/scheduler/plugins/taskorder"
 	"github.com/kai-scheduler/KAI-scheduler/pkg/scheduler/plugins/topology"
+        "github.com/kai-scheduler/KAI-scheduler/pkg/scheduler/plugins/resourceaware"
 )
 
 func InitDefaultPlugins() {
 	// Plugins for PodGroupInfos
 	framework.RegisterPluginBuilder("predicates", predicates.New)
 	framework.RegisterPluginBuilder("priority", priority.New)
+	framework.RegisterPluginBuilder("resourceaware", resourceaware.New)
 	framework.RegisterPluginBuilder("nodeplacement", nodeplacement.New)
 	framework.RegisterPluginBuilder("nominatednode", nominatednode.New)
 	framework.RegisterPluginBuilder("numa", numa.New)
@@ -79,4 +81,4 @@ func InitDefaultPlugins() {
 
 	// Always register the Job Order Plugin last.
 	framework.RegisterPluginBuilder("reflectjoborder", reflectjoborder.New)
-}
+}

pkg/scheduler/plugins/resourceaware/resourceaware.go

@@ -31,6 +31,10 @@ func (rp *resourceAwarePlugin) JobOrderFn(l, r interface{}) int {
 	lv := l.(*podgroup_info.PodGroupInfo)
 	rv := r.(*podgroup_info.PodGroupInfo)
 
+	if lv.Priority != rv.Priority {
+		return 0
+	}
+
 	lGPU := lv.GetAliveTasksRequestedGPUs()
 	rGPU := rv.GetAliveTasksRequestedGPUs()
 
@@ -53,4 +57,4 @@ func (rp *resourceAwarePlugin) JobOrderFn(l, r interface{}) int {
 	return 0
 }
 
-func (rp *resourceAwarePlugin) OnSessionClose(_ *framework.Session) {}
+func (rp *resourceAwarePlugin) OnSessionClose(_ *framework.Session) {}