gorgonia/blas.go at master · gorgonia/gorgonia

Repository navigation

History

package gorgonia

import (

"sync"

"gonum.org/v1/gonum/blas"

"gonum.org/v1/gonum/blas/gonum"

"gorgonia.org/tensor"

)

var blasdoor sync.Mutex

var whichblas BLAS

// BLAS represents all the possible implementations of BLAS.

// The default is Gonum's Native

type BLAS interface {

blas.Float32

blas.Float64

blas.Complex64

blas.Complex128

}

// only blase.Implementation() and cubone.Implementation() are batchedBLAS -

// they both batch cgo calls (and cubone batches cuda calls)

type batchedBLAS interface {

WorkAvailable() <-chan struct{}

DoWork()

Close() error

BLAS

}

// Use defines which BLAS implementation gorgonia should use.

// The default is Gonum's Native. These are the other options:

// Use(blase.Implementation())

// Use(cubone.Implementation())

// Use(cgo.Implementation)

// Note the differences in the brackets. The blase and cubone ones are functions.

func Use(b BLAS) {

// close the blast door! close the blast door!

blasdoor.Lock()

// open the blast door! open the blast door!

defer blasdoor.Unlock()

// those lines were few of the better additions to the Special Edition. There, I said it. The Special Edition is superior. Except Han still shot first in my mind.

whichblas = b

tensor.Use(b)

// TODO:

// float32

}

// WhichBLAS returns the BLAS that gorgonia uses.

func WhichBLAS() BLAS { return whichblas }

func init() {

whichblas = gonum.Implementation{}

}

FilesExpand file tree

blas.go

Latest commit

History

blas.go

File metadata and controls