base

Documentation

Index

• Constants
• func StrategyNames() []string
• func Transform(y, x *Vector, op OP)
• func Transform2(z, x, y *Vector, op OP)
• type DataType
  • func (t DataType) Size() int
  • func (t DataType) String() string
• type OP
• type Strategy
  • func ParseStrategy(s string) (*Strategy, error)
  • func (s *Strategy) Set(val string) error
  • func (s Strategy) String() string
• type Vector
  • func NewVector(count int, dtype DataType) *Vector
  • func (b *Vector) AsF32() []float32
  • func (b *Vector) AsI32() []int32
  • func (b *Vector) AsI64() []int64
  • func (b *Vector) AsI8() []int8
  • func (b *Vector) CopyFrom(c *Vector)
  • func (b *Vector) Slice(begin, end int) *Vector

Constants

const DefaultStrategy = BinaryTreeStar

Functions

func StrategyNames

func StrategyNames() []string

func Transform

func Transform(y, x *Vector, op OP)

Transform performs y[i] += x[i] for vectors y and x

func Transform2

func Transform2(z, x, y *Vector, op OP)

Transform2 performs z[i] = x[i] + y[i] for vectors z and x, y.

Types

type DataType

type DataType C.KungFu_Datatype

const (
	U8  DataType = C.KungFu_UINT8
	U16 DataType = C.KungFu_UINT16
	U32 DataType = C.KungFu_UINT32
	U64 DataType = C.KungFu_UINT64

	I8  DataType = C.KungFu_INT8
	I16 DataType = C.KungFu_INT16
	I32 DataType = C.KungFu_INT32
	I64 DataType = C.KungFu_INT64

	F16 DataType = C.KungFu_FLOAT16
	F32 DataType = C.KungFu_FLOAT
	F64 DataType = C.KungFu_DOUBLE
)

func (DataType) Size

func (t DataType) Size() int

func (DataType) String

func (t DataType) String() string

type OP

type OP C.KungFu_Op

const (
	SUM  OP = C.KungFu_SUM
	MIN  OP = C.KungFu_MIN
	MAX  OP = C.KungFu_MAX
	PROD OP = C.KungFu_PROD
)

type Strategy

type Strategy C.KungFu_Strategy

const (
	Star                Strategy = C.KungFu_Star
	Ring                Strategy = C.KungFu_Ring
	Clique              Strategy = C.KungFu_Clique
	Tree                Strategy = C.KungFu_Tree
	BinaryTree          Strategy = C.KungFu_BinaryTree
	BinaryTreeStar      Strategy = C.KungFu_BinaryTreeStar
	MultiBinaryTreeStar Strategy = C.KungFu_MultiBinaryTreeStar
	Auto                Strategy = C.KungFu_AUTO
)

func ParseStrategy

func ParseStrategy(s string) (*Strategy, error)

func (*Strategy) Set

func (s *Strategy) Set(val string) error

Set implements flags.Value::Set

func (Strategy) String

func (s Strategy) String() string

type Vector

type Vector struct {
	Data  []byte
	Count int
	Type  DataType
}

func NewVector

func NewVector(count int, dtype DataType) *Vector

func (*Vector) AsF32

func (b *Vector) AsF32() []float32

func (*Vector) AsI32

func (b *Vector) AsI32() []int32

func (*Vector) AsI64

func (b *Vector) AsI64() []int64

func (*Vector) AsI8

func (b *Vector) AsI8() []int8

func (*Vector) CopyFrom

func (b *Vector) CopyFrom(c *Vector)

func (*Vector) Slice

func (b *Vector) Slice(begin, end int) *Vector

Slice returns a new Vector that points to a subset of the original Vector. 0 <= begin < end <= count - 1