cuda

Documentation

Index

• Variables
• func GetBnBpropKernel(dtype base.Dtype, threads int, computeCapability [2]int) (string, string)
• func GetBnFpropKernel(dtype base.Dtype, threads int, computeCapability [2]int) (string, string)
• func GetConvKernel(dtype base.Dtype, filterSize int, bsum bool, operation string, ...) (string, string)
• func MapStringToFunc(funcname string, dtype base.Dtype, computeCapability [2]int) (string, string)

Variables

var (
	// data
	InitRandFunc      = initRandFunc
	InitRandRoundFunc = initRandRoundFunc
	FinishRandFunc    = finishRandFunc
	CommonKepler      = commonKepler
	CommonUrandGen    = commonUrandGen
	CommonFrand       = commonFrand
	CommonRound       = commonRound
	CommonFp16toFp32  = commonFp16toFp32
	CommonMaxAbs      = commonMaxAbs
	EwTypes           = ewTypes
	EwStringsRound    = ewStringsRound
	// functions
	Format = format
)

Functions

func GetBnBpropKernel

func GetBnBpropKernel(dtype base.Dtype, threads int, computeCapability [2]int) (string, string)

func GetBnFpropKernel

func GetBnFpropKernel(dtype base.Dtype, threads int, computeCapability [2]int) (string, string)

func GetConvKernel

func GetConvKernel(
	dtype base.Dtype,
	filterSize int,
	bsum bool,
	operation string,
	filterBoundsCheck bool,
	debug bool) (string, string)

func MapStringToFunc

func MapStringToFunc(
	funcname string,
	dtype base.Dtype,
	computeCapability [2]int) (string, string)

Helper function that converts string function names to function calls