Skip to content

Commit 4553073

Browse files
committed
1- Do not count maxed queues for portion calculations
2- Fix unitests by adding UID to struct definition
1 parent dfde852 commit 4553073

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

pkg/scheduler/plugins/proportion/resource_division/resource_division.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,11 @@ func divideUpToFairShare(totalResourceAmount, kValue float64, queues map[common_
182182
// queue is satisfied, no need to give it more resources
183183
continue
184184
}
185+
if share.MaxAllowed != commonconstants.UnlimitedResourceQuantity &&
186+
share.MaxAllowed <= share.FairShare {
187+
// queue is satisfied, no need to give it more resources
188+
continue
189+
}
185190

186191
// Normalize queue over quota weight
187192
nWeight := share.OverQuotaWeight / totalWeights

pkg/scheduler/plugins/proportion/resource_division/resource_division_test.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1414,6 +1414,7 @@ var _ = Describe("Proportion", func() {
14141414
"Request more than total with different Over Quota": {
14151415
queues: map[common_info.QueueID]*rs.QueueAttributes{
14161416
"1": {
1417+
UID: "1",
14171418
QueueResourceShare: rs.QueueResourceShare{
14181419
GPU: rs.ResourceShare{
14191420
Deserved: 250,
@@ -1442,6 +1443,7 @@ var _ = Describe("Proportion", func() {
14421443
},
14431444
},
14441445
"2": {
1446+
UID: "2",
14451447
QueueResourceShare: rs.QueueResourceShare{
14461448
GPU: rs.ResourceShare{
14471449
Deserved: 150,
@@ -1507,6 +1509,7 @@ var _ = Describe("Proportion", func() {
15071509
"Request more than total with different Over Quota and max allowed": {
15081510
queues: map[common_info.QueueID]*rs.QueueAttributes{
15091511
"1": {
1512+
UID: "1",
15101513
QueueResourceShare: rs.QueueResourceShare{
15111514
GPU: rs.ResourceShare{
15121515
Deserved: 250,
@@ -1535,6 +1538,7 @@ var _ = Describe("Proportion", func() {
15351538
},
15361539
},
15371540
"2": {
1541+
UID: "2",
15381542
QueueResourceShare: rs.QueueResourceShare{
15391543
GPU: rs.ResourceShare{
15401544
Deserved: 150,
@@ -1736,6 +1740,7 @@ var _ = Describe("Proportion", func() {
17361740
"divide resource between queues with different over quota": {
17371741
queues: map[common_info.QueueID]*rs.QueueAttributes{
17381742
"1": {
1743+
UID: "1",
17391744
QueueResourceShare: rs.QueueResourceShare{
17401745
GPU: rs.ResourceShare{
17411746
Deserved: 0,
@@ -1764,6 +1769,7 @@ var _ = Describe("Proportion", func() {
17641769
},
17651770
},
17661771
"2": {
1772+
UID: "2",
17671773
QueueResourceShare: rs.QueueResourceShare{
17681774
GPU: rs.ResourceShare{
17691775
Deserved: 0,
@@ -1792,6 +1798,7 @@ var _ = Describe("Proportion", func() {
17921798
},
17931799
},
17941800
"3": {
1801+
UID: "3",
17951802
QueueResourceShare: rs.QueueResourceShare{
17961803
GPU: rs.ResourceShare{
17971804
Deserved: 0,

0 commit comments

Comments
 (0)