relu

Documentation

Overview

Package relu implements the relu algorithm as a benchmark.

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 KernelArgs

Types

type Benchmark

type Benchmark struct {
	Length int
	// contains filtered or unexported fields
}

Benchmark defines a benchmark

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark returns a 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 KernelArgs

type KernelArgs struct {
	Count               uint32
	Padding             uint32
	Input               driver.Ptr
	Output              driver.Ptr
	HiddenGlobalOffsetX int64
	HiddenGlobalOffsetY int64
	HiddenGlobalOffsetZ int64
}

KernelArgs defines kernel arguments