zmatrix

Documentation

Index

• func FreeMatrix(mat MatrixFloat64)
• func MatrixFloat64Mul(a []float64, b []float64, c []float64, grid []int32, block []int32, index int, ...)
• func VerifyFloat64RandResult(a *MatrixFloat64, b *MatrixFloat64, result *MatrixFloat64) bool
• func VerifyGoNumResult(a []float64, b *MatrixFloat64, result []float64, begin int) bool
• func VerifyRandResult(a *Matrix, b *Matrix, result *Matrix) bool
• func VerifyResult(a *Matrix, b *Matrix) bool
• type Matrix
  • func (m *Matrix) Rand()
• type MatrixFloat64
  • func GpuMatrixFloat64Mul(a *MatrixFloat64, b *MatrixFloat64, index int) (*MatrixFloat64, error)
  • func NewMatrixFloat64(row, col int64) MatrixFloat64
  • func (m *MatrixFloat64) Rand()

Functions

func FreeMatrix

func FreeMatrix(mat MatrixFloat64)

func MatrixFloat64Mul

func MatrixFloat64Mul(a []float64, b []float64, c []float64, grid []int32, block []int32, index int, m, n, k int64)

func VerifyFloat64RandResult

func VerifyFloat64RandResult(a *MatrixFloat64, b *MatrixFloat64, result *MatrixFloat64) bool

func VerifyGoNumResult

func VerifyGoNumResult(a []float64, b *MatrixFloat64, result []float64, begin int) bool

func VerifyRandResult

func VerifyRandResult(a *Matrix, b *Matrix, result *Matrix) bool

func VerifyResult

func VerifyResult(a *Matrix, b *Matrix) bool

Types

type Matrix

type Matrix struct {
	Row  int //行
	Col  int //列
	Data *[maxInt]int32
}

func (*Matrix) Rand

func (m *Matrix) Rand()

type MatrixFloat64

type MatrixFloat64 struct {
	Row int64 //行
	Col int64 //列

	Data []float64
	// contains filtered or unexported fields
}

func GpuMatrixFloat64Mul

func GpuMatrixFloat64Mul(a *MatrixFloat64, b *MatrixFloat64, index int) (*MatrixFloat64, error)

func NewMatrixFloat64

func NewMatrixFloat64(row, col int64) MatrixFloat64

func (*MatrixFloat64) Rand

func (m *MatrixFloat64) Rand()