atax

Documentation

Overview

Package atax implements the ATAX benchmark from Polybench.

Index

• type Benchmark
  • func NewBenchmark(driver *driver.Driver) *Benchmark
  • func (b *Benchmark) Run()
  • func (b *Benchmark) SelectGPU(gpus []int)
  • func (b *Benchmark) SetUnifiedMemory()
  • func (b *Benchmark) Verify()
• type Kernel1Args
• type Kernel2Args

Types

type Benchmark

type Benchmark struct {
	NX, NY int
	// contains filtered or unexported fields
}

Benchmark defines a benchmark

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark makes a new benchmark

func (*Benchmark) Run

func (b *Benchmark) Run()

Run runs

func (*Benchmark) SelectGPU

func (b *Benchmark) SelectGPU(gpus []int)

SelectGPU selects GPU

func (*Benchmark) SetUnifiedMemory

func (b *Benchmark) SetUnifiedMemory()

SetUnifiedMemory uses Unified Memory

func (*Benchmark) Verify

func (b *Benchmark) Verify()

Verify verifies

type Kernel1Args

type Kernel1Args struct {
	A   driver.Ptr
	X   driver.Ptr
	Tmp driver.Ptr
	NX  int32
	NY  int32
}

Kernel1Args list first set of kernel arguments

type Kernel2Args

type Kernel2Args struct {
	A      driver.Ptr
	Y      driver.Ptr
	Tmp    driver.Ptr
	NX, NY int32
}

Kernel2Args list second set of kernel arguments