spmv

Documentation

Overview

Package spmv include the benchmark of sparse matrix-vector matiplication.

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 {
	Dim      int32
	Sparsity float64
	// contains filtered or unexported fields
}

Benchmark set up test parameters

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark creates a new benchmark

func (*Benchmark) Run

func (b *Benchmark) Run()

Run runs the benchmark

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 results

type KernelArgs

type KernelArgs struct {
	Val           driver.GPUPtr
	Vec           driver.GPUPtr
	Cols          driver.GPUPtr
	RowDelimiters driver.GPUPtr
	Dim           int32
	// VecWidth            int32
	// PartialSums         driver.LocalPtr
	Padding             int32
	Out                 driver.GPUPtr
	HiddenGlobalOffsetX int64
	HiddenGlobalOffsetY int64
	HiddenGlobalOffsetZ int64
}

KernelArgs sets up kernel arguments