job

Documentation

Index

• type Envs
• type GPUPool
  • func NewGPUPool(n int) *GPUPool
  • func (p *GPUPool) Get() int
  • func (p *GPUPool) Put(id int)
• type Job
  • func (j Job) CreateAllProcs(pl plan.PeerList) []Proc
  • func (j Job) CreateProcs(pl plan.PeerList, host uint32) []Proc
  • func (j Job) NewProc(peer plan.PeerID, gpuID int, initClusterVersion int, pl plan.PeerList) Proc
• type Proc
  • func (p Proc) Cmd() *exec.Cmd
  • func (p Proc) Script() string

Types

type Envs

type Envs map[string]string

type GPUPool

type GPUPool struct {
	sync.Mutex
	// contains filtered or unexported fields
}

GPUPool manages GPU ids

func NewGPUPool

func NewGPUPool(n int) *GPUPool

NewGPUPool create a new GPUPool of given size

func (*GPUPool) Get

func (p *GPUPool) Get() int

Get returns the smallest GPU id that is available

func (*GPUPool) Put

func (p *GPUPool) Put(id int)

Put puts an GPU id back to the pool

type Job

type Job struct {
	ConfigServer string
	Strategy     kb.Strategy
	Parent       plan.PeerID
	HostList     plan.HostList
	PortRange    plan.PortRange
	Prog         string
	Args         []string
	LogDir       string

	AllowNVLink bool
}

func (Job) CreateAllProcs

func (j Job) CreateAllProcs(pl plan.PeerList) []Proc

func (Job) CreateProcs

func (j Job) CreateProcs(pl plan.PeerList, host uint32) []Proc

func (Job) NewProc

func (j Job) NewProc(peer plan.PeerID, gpuID int, initClusterVersion int, pl plan.PeerList) Proc

type Proc

type Proc struct {
	Name    string
	Prog    string
	Args    []string
	Envs    Envs
	IPv4    uint32
	PubAddr string
	LogDir  string
}

Proc represents a process

func (Proc) Cmd

func (p Proc) Cmd() *exec.Cmd

func (Proc) Script

func (p Proc) Script() string