YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-13 07:46:43 +00:00
Code Issues Releases Wiki Activity

add GPU share support

Browse Source
This commit is contained in:
Newnius
2019-05-13 16:31:26 +08:00
parent 412bf9b8c7
commit 4d14a93b88
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/AllocatorFIFO.go
View File

@@ -85,7 +85,7 @@ func (allocator *AllocatorFIFO) requestResource(task Task) NodeStatus {
for id, node := range pool.nodes { for id, node := range pool.nodes {
var available []GPUStatus var available []GPUStatus
for _, status := range node.Status { for _, status := range node.Status {
if status.MemoryAllocated == 0 { if status.MemoryTotal-status.MemoryAllocated >= task.MemoryGPU {
available = append(available, status) available = append(available, status)
} }
} }
@@ -97,7 +97,8 @@ func (allocator *AllocatorFIFO) requestResource(task Task) NodeStatus {
for i := range res.Status { for i := range res.Status {
for j := range node.Status { for j := range node.Status {
if res.Status[i].UUID == node.Status[j].UUID { if res.Status[i].UUID == node.Status[j].UUID {
node.Status[j].MemoryAllocated = task.MemoryGPU node.Status[j].MemoryAllocated += task.MemoryGPU
res.Status[i].MemoryTotal = task.MemoryGPU
} }
} }
} }
@@ -114,7 +115,7 @@ func (allocator *AllocatorFIFO) returnResource(agent NodeStatus) {
for _, gpu := range agent.Status { for _, gpu := range agent.Status {
for j := range nodes.Status { for j := range nodes.Status {
if gpu.UUID == nodes.Status[j].UUID { if gpu.UUID == nodes.Status[j].UUID {
nodes.Status[j].MemoryAllocated = 0 nodes.Status[j].MemoryAllocated -= gpu.MemoryTotal
} }
} }
} }