gpu

Documentation

Index

• type BestFitMemory
  • func (s *BestFitMemory) OnChooseGPU(gpuMemReq uint32, gpuNodes []*models.GPUNode, numWorkerPerGPU uint32) (string, uint32, int, *models.GPUNode)
• type FirstFitMemory
  • func (s *FirstFitMemory) OnChooseGPU(gpuMemReq uint32, gpuNodes []*models.GPUNode, numWorkerPerGPU uint32) (string, uint32, int, *models.GPUNode)
• type GPUSelection

Types

type BestFitMemory

type BestFitMemory struct{}

BestFitMemory struct for bestfit memory policy

func (*BestFitMemory) OnChooseGPU

func (s *BestFitMemory) OnChooseGPU(gpuMemReq uint32, gpuNodes []*models.GPUNode, numWorkerPerGPU uint32) (string, uint32, int, *models.GPUNode)

OnChooseGPU implements best fit policy regarding memory

type FirstFitMemory

type FirstFitMemory struct{}

FirstFitMemory struct for first fit memory policy

func (*FirstFitMemory) OnChooseGPU

func (s *FirstFitMemory) OnChooseGPU(gpuMemReq uint32, gpuNodes []*models.GPUNode, numWorkerPerGPU uint32) (string, uint32, int, *models.GPUNode)

OnChooseGPU implements first fit policy regarding memory

type GPUSelection

type GPUSelection interface {
	OnChooseGPU(gpuMemReq uint32, gpuNodes []*models.GPUNode, numWorkerPerGPU uint32) (string, uint32, *models.GPUNode)
}

GPUSelection is the GPU scheduling policy interface