bench

Documentation

Index

• Variables
• func IsKnownTool(tool Tool) bool
• type BenchResult
  • func Run(ctx context.Context, runner execRunner, tool Tool, duration time.Duration) (*BenchResult, error)
  • func RunWithOptions(ctx context.Context, runner execRunnerV2, tool Tool, opts Options) (*BenchResult, error)
• type ExecResult
• type Options
• type Tool

Variables

var ErrToolNotInstalled = errors.New("benchmark tool not installed")

Functions

func IsKnownTool

func IsKnownTool(tool Tool) bool

Types

type BenchResult

type BenchResult struct {
	Tool       string
	Duration   time.Duration
	Throughput float64
	Unit       string
	RawLog     string
	GDRDMA     string `json:"gdrdma,omitempty"`
}

func Run

func Run(ctx context.Context, runner execRunner, tool Tool, duration time.Duration) (*BenchResult, error)

func RunWithOptions

func RunWithOptions(ctx context.Context, runner execRunnerV2, tool Tool, opts Options) (*BenchResult, error)

type ExecResult

type ExecResult struct {
	Stdout []byte
	Stderr []byte
}

type Options

type Options struct {
	Duration  time.Duration
	Server    string
	UseCUDA   *int
	GPUs      int
	NCCLDebug bool
	ExtraArgs []string
}

type Tool

type Tool string

const (
	ToolGPUBurn       Tool = "gpu-burn"
	ToolNVBandwidth   Tool = "nvbandwidth"
	ToolBandwidthTest Tool = "bandwidthTest"
	ToolPerftest      Tool = "perftest"
	ToolNCCLTests     Tool = "nccl-tests"
)