cublas

package
v0.0.0-...-2b09442 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Nov 19, 2020 License: Apache-2.0 Imports: 4 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Create

func Create(p *Handle) error

func Destroy

func Destroy(p Handle) error

func SetStream

func SetStream(handle Handle, stream cuda.Stream) error

func Sgemm

func Sgemm(
	handle Handle, opA, opB Operation, m, n, k int, alpha *float32,
	a unsafe.Pointer, lda int, b unsafe.Pointer, ldb int, beta *float32, c unsafe.Pointer, ldc int) error

Types

type Handle

type Handle C.cublasHandle_t

type Operation

type Operation C.cublasOperation_t
var OpN Operation = C.CUBLAS_OP_N
var OpT Operation = C.CUBLAS_OP_T

type Status

type Status C.cublasStatus_t

func (Status) Error

func (e Status) Error() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL