vm

Documentation

Overview

Package vm provides the models for address translations

Index

• type Page
• type PageMigrationInfo
• type PageMigrationReqToDriver
  • func NewPageMigrationReqToDriver(time akita.VTimeInSec, src, dst akita.Port) *PageMigrationReqToDriver
  • func (m *PageMigrationReqToDriver) Meta() *akita.MsgMeta
• type PageMigrationRspFromDriver
  • func NewPageMigrationRspFromDriver(time akita.VTimeInSec, src, dst akita.Port) *PageMigrationRspFromDriver
  • func (m *PageMigrationRspFromDriver) Meta() *akita.MsgMeta
• type PageTable
  • func NewPageTable(log2PageSize uint64) PageTable
• type TLBTracer
  • func NewTLBTracer(w io.Writer) *TLBTracer
  • func (t *TLBTracer) Func(ctx *akita.HookCtx)
• type TranslationReq
  • func (r *TranslationReq) Meta() *akita.MsgMeta
• type TranslationReqBuilder
  • func (b TranslationReqBuilder) Build() *TranslationReq
  • func (b TranslationReqBuilder) WithDst(dst akita.Port) TranslationReqBuilder
  • func (b TranslationReqBuilder) WithGPUID(gpuID uint64) TranslationReqBuilder
  • func (b TranslationReqBuilder) WithPID(pid ca.PID) TranslationReqBuilder
  • func (b TranslationReqBuilder) WithSendTime(t akita.VTimeInSec) TranslationReqBuilder
  • func (b TranslationReqBuilder) WithSrc(src akita.Port) TranslationReqBuilder
  • func (b TranslationReqBuilder) WithVAddr(vAddr uint64) TranslationReqBuilder
• type TranslationRsp
  • func (r *TranslationRsp) Meta() *akita.MsgMeta
• type TranslationRspBuilder
  • func (b TranslationRspBuilder) Build() *TranslationRsp
  • func (b TranslationRspBuilder) WithDst(dst akita.Port) TranslationRspBuilder
  • func (b TranslationRspBuilder) WithPage(page Page) TranslationRspBuilder
  • func (b TranslationRspBuilder) WithRspTo(rspTo string) TranslationRspBuilder
  • func (b TranslationRspBuilder) WithSendTime(t akita.VTimeInSec) TranslationRspBuilder
  • func (b TranslationRspBuilder) WithSrc(src akita.Port) TranslationRspBuilder

Types

type Page

type Page struct {
	PID         ca.PID
	PAddr       uint64
	VAddr       uint64
	PageSize    uint64
	Valid       bool
	GPUID       uint64
	Unified     bool
	IsMigrating bool
	IsPinned    bool
}

A Page is an entry in the page table, maintaining the information about how to translate a virtual address to a physical address

type PageMigrationInfo

type PageMigrationInfo struct {
	GpuReqToVAddrMap map[uint64][]uint64
}

type PageMigrationReqToDriver

type PageMigrationReqToDriver struct {
	akita.MsgMeta

	StartTime         akita.VTimeInSec
	EndTime           akita.VTimeInSec
	MigrationInfo     *PageMigrationInfo
	CurrAccessingGPUs []uint64
	PID               ca.PID
	CurrPageHostGPU   uint64
	PageSize          uint64
	RespondToTop      bool
}

PageMigrationReqToDriver is a req to driver from MMU to start page migration process

func NewPageMigrationReqToDriver

func NewPageMigrationReqToDriver(
	time akita.VTimeInSec,
	src, dst akita.Port,
) *PageMigrationReqToDriver

func (*PageMigrationReqToDriver) Meta

func (m *PageMigrationReqToDriver) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type PageMigrationRspFromDriver

type PageMigrationRspFromDriver struct {
	akita.MsgMeta

	StartTime akita.VTimeInSec
	EndTime   akita.VTimeInSec
	VAddr     []uint64
	RspToTop  bool
}

PageMigrationRspFromDriver is a rsp from driver to MMU marking completion of migration

func NewPageMigrationRspFromDriver

func NewPageMigrationRspFromDriver(
	time akita.VTimeInSec,
	src, dst akita.Port,
) *PageMigrationRspFromDriver

func (*PageMigrationRspFromDriver) Meta

func (m *PageMigrationRspFromDriver) Meta() *akita.MsgMeta

Meta returns the meta data associated with the message.

type PageTable

type PageTable interface {
	Insert(page Page)
	Remove(pid ca.PID, vAddr uint64)
	Find(pid ca.PID, Addr uint64) (Page, bool)
	Update(page Page)
}

A PageTable holds the a list of pages.

func NewPageTable

func NewPageTable(log2PageSize uint64) PageTable

type TLBTracer

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

A TLBTracer write logs for what happened in a TLB

func NewTLBTracer

func NewTLBTracer(w io.Writer) *TLBTracer

NewTLBTracer produce a new TLBTracer, injecting the dependency of a writer.

func (*TLBTracer) Func

func (t *TLBTracer) Func(ctx *akita.HookCtx)

Func prints the tlb trace information.

type TranslationReq

type TranslationReq struct {
	akita.MsgMeta
	VAddr uint64
	PID   ca.PID
	GPUID uint64
}

A TranslationReq asks the receiver component to translate the request.

func (*TranslationReq) Meta

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

Meta returns the meta data associated with the message.

type TranslationReqBuilder

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

TranslationReqBuilder can build translation requests

func (TranslationReqBuilder) Build

func (b TranslationReqBuilder) Build() *TranslationReq

Build creats a new TranslationReq

func (TranslationReqBuilder) WithDst

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

WithDst sets the destination of the request to build.

func (TranslationReqBuilder) WithGPUID

func (b TranslationReqBuilder) WithGPUID(gpuID uint64) TranslationReqBuilder

WithGPUID sets the GPU ID of the request to build.

func (TranslationReqBuilder) WithPID

func (b TranslationReqBuilder) WithPID(pid ca.PID) TranslationReqBuilder

WithPID sets the virtual address of the request to build.

func (TranslationReqBuilder) WithSendTime

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

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

func (TranslationReqBuilder) WithSrc

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

WithSrc sets the source of the request to build.

func (TranslationReqBuilder) WithVAddr

func (b TranslationReqBuilder) WithVAddr(vAddr uint64) TranslationReqBuilder

WithVAddr sets the virtual address of the request to build.

type TranslationRsp

type TranslationRsp struct {
	akita.MsgMeta
	RespondTo string // The ID of the request it replies
	Page      Page
}

A TranslationRsp is the respond for a TranslationReq. It carries the physical address.

func (*TranslationRsp) Meta

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

Meta returns the meta data associated with the message.

type TranslationRspBuilder

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

TranslationRspBuilder can build translation requests

func (TranslationRspBuilder) Build

func (b TranslationRspBuilder) Build() *TranslationRsp

Build creats a new TranslationRsp

func (TranslationRspBuilder) WithDst

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

WithDst sets the destination of the respond to build.

func (TranslationRspBuilder) WithPage

func (b TranslationRspBuilder) WithPage(page Page) TranslationRspBuilder

WithPage sets the page of the respond to build.

func (TranslationRspBuilder) WithRspTo

func (b TranslationRspBuilder) WithRspTo(rspTo string) TranslationRspBuilder

WithRspTo sets the request ID of the respond to build.

func (TranslationRspBuilder) WithSendTime

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

WithSendTime sets the send time of the message to build.

func (TranslationRspBuilder) WithSrc

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

WithSrc sets the source of the respond to build.