kmeans

Documentation

Overview

Package kmeans implements the Kmeans benchmark form Hetero-Mark.

Index

• type Benchmark
  • func NewBenchmark(driver *driver.Driver) *Benchmark
  • func (b *Benchmark) Run()
  • func (b *Benchmark) SelectGPU(gpuIDs []int)
  • func (b *Benchmark) SetUnifiedMemory()
  • func (b *Benchmark) Verify()
• type ComputeArgs
• type SwapArgs

Types

type Benchmark

type Benchmark struct {
	NumClusters int
	NumPoints   int
	NumFeatures int
	MaxIter     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(gpuIDs []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 ComputeArgs

type ComputeArgs struct {
	Feature             driver.Ptr
	Clusters            driver.Ptr
	Membership          driver.Ptr
	NPoints             int32
	NClusters           int32
	NFeatures           int32
	Offset              int32
	Size                int32
	Padding             int32
	HiddenGlobalOffsetX int64
	HiddenGlobalOffsetY int64
	HiddenGlobalOffsetZ int64
}

ComputeArgs defines arguments

type SwapArgs

type SwapArgs struct {
	Feature             driver.Ptr
	FeatureSwap         driver.Ptr
	NPoints             int32
	NFeatures           int32
	HiddenGlobalOffsetX int64
	HiddenGlobalOffsetY int64
	HiddenGlobalOffsetZ int64
}

SwapArgs defines arguments