rdma

Documentation

Overview

Package rdma provides the implementation of an RDMA engine.

Index

• type DrainReq
  • func (r *DrainReq) Meta() *akita.MsgMeta
• type DrainReqBuilder
  • func (b DrainReqBuilder) Build() *DrainReq
  • func (b DrainReqBuilder) WithDst(dst akita.Port) DrainReqBuilder
  • func (b DrainReqBuilder) WithSendTime(t akita.VTimeInSec) DrainReqBuilder
  • func (b DrainReqBuilder) WithSrc(src akita.Port) DrainReqBuilder
• type DrainRsp
  • func (r *DrainRsp) Meta() *akita.MsgMeta
• type DrainRspBuilder
  • func (b DrainRspBuilder) Build() *DrainRsp
  • func (b DrainRspBuilder) WithDst(dst akita.Port) DrainRspBuilder
  • func (b DrainRspBuilder) WithSendTime(t akita.VTimeInSec) DrainRspBuilder
  • func (b DrainRspBuilder) WithSrc(src akita.Port) DrainRspBuilder
• type Engine
  • func NewEngine(name string, engine akita.Engine, localModules cache.LowModuleFinder, ...) *Engine
  • func (e *Engine) SetFreq(freq akita.Freq)
  • func (e *Engine) SetLocalModuleFinder(lmf cache.LowModuleFinder)
  • func (e *Engine) Tick(now akita.VTimeInSec) bool
• type RestartReq
  • func (r *RestartReq) Meta() *akita.MsgMeta
• type RestartReqBuilder
  • func (b RestartReqBuilder) Build() *RestartReq
  • func (b RestartReqBuilder) WithDst(dst akita.Port) RestartReqBuilder
  • func (b RestartReqBuilder) WithSendTime(t akita.VTimeInSec) RestartReqBuilder
  • func (b RestartReqBuilder) WithSrc(src akita.Port) RestartReqBuilder
• type RestartRsp
  • func (r *RestartRsp) Meta() *akita.MsgMeta
• type RestartRspBuilder
  • func (b RestartRspBuilder) Build() *RestartRsp
  • func (b RestartRspBuilder) WithDst(dst akita.Port) RestartRspBuilder
  • func (b RestartRspBuilder) WithSendTime(t akita.VTimeInSec) RestartRspBuilder
  • func (b RestartRspBuilder) WithSrc(src akita.Port) RestartRspBuilder

Types

type DrainReq

type DrainReq struct {
	akita.MsgMeta
}

DrainReq asks the rdma to stop processing requests from L1 while allowing pending requests to L2 to complete

func (*DrainReq) Meta

func (r *DrainReq) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type DrainReqBuilder

type DrainReqBuilder struct {
	// contains filtered or unexported fields
}

DrainReqBuilder can build RDMA drain requests

func (DrainReqBuilder) Build

func (b DrainReqBuilder) Build() *DrainReq

Build creats a new DrainReq

func (DrainReqBuilder) WithDst

func (b DrainReqBuilder) WithDst(dst akita.Port) DrainReqBuilder

WithDst sets the destination of the request to build.

func (DrainReqBuilder) WithSendTime

func (b DrainReqBuilder) WithSendTime(
	t akita.VTimeInSec,
) DrainReqBuilder

WithSendTime sets the send time of the request to build.:w

func (DrainReqBuilder) WithSrc

func (b DrainReqBuilder) WithSrc(src akita.Port) DrainReqBuilder

WithSrc sets the source of the request to build.

type DrainRsp

type DrainRsp struct {
	akita.MsgMeta
}

DrainRsp is a drain complete response to a RDMA Drain Req

func (*DrainRsp) Meta

func (r *DrainRsp) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type DrainRspBuilder

type DrainRspBuilder struct {
	// contains filtered or unexported fields
}

DrainRspBuilder can build RDMA drain responses

func (DrainRspBuilder) Build

func (b DrainRspBuilder) Build() *DrainRsp

Build creats a new RDMADrainRsp

func (DrainRspBuilder) WithDst

func (b DrainRspBuilder) WithDst(dst akita.Port) DrainRspBuilder

WithDst sets the destination of the request to build.

func (DrainRspBuilder) WithSendTime

func (b DrainRspBuilder) WithSendTime(
	t akita.VTimeInSec,
) DrainRspBuilder

WithSendTime sets the send time of the request to build

func (DrainRspBuilder) WithSrc

func (b DrainRspBuilder) WithSrc(src akita.Port) DrainRspBuilder

WithSrc sets the source of the request to build.

type Engine

type Engine struct {
	*akita.TickingComponent

	ToOutside akita.Port

	ToL1 akita.Port
	ToL2 akita.Port

	CtrlPort akita.Port

	RemoteRDMAAddressTable cache.LowModuleFinder
	// contains filtered or unexported fields
}

An Engine is a component that helps one GPU to access the memory on another GPU

func NewEngine

func NewEngine(
	name string,
	engine akita.Engine,
	localModules cache.LowModuleFinder,
	remoteModules cache.LowModuleFinder,
) *Engine

NewEngine creates new engine

func (*Engine) SetFreq

func (e *Engine) SetFreq(freq akita.Freq)

SetFreq sets freq

func (*Engine) SetLocalModuleFinder

func (e *Engine) SetLocalModuleFinder(lmf cache.LowModuleFinder)

SetLocalModuleFinder sets the table to lookup for local data.

func (*Engine) Tick

func (e *Engine) Tick(now akita.VTimeInSec) bool

Tick checks if make progress

type RestartReq

type RestartReq struct {
	akita.MsgMeta
}

RestartReq is a message to allow rdma to continue processing reqs from L1

func (*RestartReq) Meta

func (r *RestartReq) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type RestartReqBuilder

type RestartReqBuilder struct {
	// contains filtered or unexported fields
}

RestartReqBuilder can build RDMA restart req

func (RestartReqBuilder) Build

func (b RestartReqBuilder) Build() *RestartReq

Build creats a new RDMADrainRsp

func (RestartReqBuilder) WithDst

func (b RestartReqBuilder) WithDst(dst akita.Port) RestartReqBuilder

WithDst sets the destination of the request to build.

func (RestartReqBuilder) WithSendTime

func (b RestartReqBuilder) WithSendTime(
	t akita.VTimeInSec,
) RestartReqBuilder

WithSendTime sets the send time of the request to build

func (RestartReqBuilder) WithSrc

func (b RestartReqBuilder) WithSrc(src akita.Port) RestartReqBuilder

WithSrc sets the source of the request to build.

type RestartRsp

type RestartRsp struct {
	akita.MsgMeta
}

RestartRsp is a drain complete response to a RDMA Drain Req

func (*RestartRsp) Meta

func (r *RestartRsp) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type RestartRspBuilder

type RestartRspBuilder struct {
	// contains filtered or unexported fields
}

RestartRspBuilder can build RDMA drain responses

func (RestartRspBuilder) Build

func (b RestartRspBuilder) Build() *RestartRsp

Build creats a new RDMADrainRsp

func (RestartRspBuilder) WithDst

func (b RestartRspBuilder) WithDst(dst akita.Port) RestartRspBuilder

WithDst sets the destination of the request to build.

func (RestartRspBuilder) WithSendTime

func (b RestartRspBuilder) WithSendTime(
	t akita.VTimeInSec,
) RestartRspBuilder

WithSendTime sets the send time of the request to build

func (RestartRspBuilder) WithSrc

func (b RestartRspBuilder) WithSrc(src akita.Port) RestartRspBuilder

WithSrc sets the source of the request to build.