Foreign.CUDA.BLAS.Context

Context management

newtype Handle

An opaque handle to the cuBLAS library context, which is passed to all library function calls.

http://docs.nvidia.com/cuda/cublas/index.html#cublashandle_t

Constructors

Handle

Fields

• useHandle :: Ptr ()

create :: IO Handle

This function initializes the CUBLAS library and creates a handle to an opaque structure holding the CUBLAS library context. It allocates hardware resources on the host and device and must be called prior to making any other CUBLAS library calls.

http://docs.nvidia.com/cuda/cublas/index.html#cublascreate

destroy :: Handle -> IO ()

This function releases hardware resources used by the CUBLAS library. The release of GPU resources may be deferred until the application exits. This function is usually the last call with a particular handle to the CUBLAS library.

http://docs.nvidia.com/cuda/cublas/index.html#cublasdestroy

Utilities

data PointerMode

For functions which take scalar value arguments, determines whether those values are passed by reference on the host or device.

http://docs.nvidia.com/cuda/cublas/index.html#cublaspointermode_t

Constructors

Host

Device

Instances

Enum PointerMode

Methods

succ :: PointerMode -> PointerMode

pred :: PointerMode -> PointerMode

toEnum :: Int -> PointerMode

fromEnum :: PointerMode -> Int

enumFrom :: PointerMode -> [PointerMode]

enumFromThen :: PointerMode -> PointerMode -> [PointerMode]

enumFromTo :: PointerMode -> PointerMode -> [PointerMode]

enumFromThenTo :: PointerMode -> PointerMode -> PointerMode -> [PointerMode]

Eq PointerMode

Methods

(==) :: PointerMode -> PointerMode -> Bool

(/=) :: PointerMode -> PointerMode -> Bool

Show PointerMode

Methods

showsPrec :: Int -> PointerMode -> ShowS

show :: PointerMode -> String

showList :: [PointerMode] -> ShowS

data AtomicsMode

Determines whether cuBLAS routines can alternate implementations which make use of atomic instructions.

http://docs.nvidia.com/cuda/cublas/index.html#cublasatomicsmode_t

Constructors

NotAllowed

Allowed

Instances

Enum AtomicsMode

Methods

succ :: AtomicsMode -> AtomicsMode

pred :: AtomicsMode -> AtomicsMode

toEnum :: Int -> AtomicsMode

fromEnum :: AtomicsMode -> Int

enumFrom :: AtomicsMode -> [AtomicsMode]

enumFromThen :: AtomicsMode -> AtomicsMode -> [AtomicsMode]

enumFromTo :: AtomicsMode -> AtomicsMode -> [AtomicsMode]

enumFromThenTo :: AtomicsMode -> AtomicsMode -> AtomicsMode -> [AtomicsMode]

Eq AtomicsMode

Methods

(==) :: AtomicsMode -> AtomicsMode -> Bool

(/=) :: AtomicsMode -> AtomicsMode -> Bool

Show AtomicsMode

Methods

showsPrec :: Int -> AtomicsMode -> ShowS

show :: AtomicsMode -> String

showList :: [AtomicsMode] -> ShowS

data MathMode

Enum for default math mode / tensor math mode

Constructors

DefaultMath

TensorMath

Instances

Enum MathMode

Methods

succ :: MathMode -> MathMode

pred :: MathMode -> MathMode

toEnum :: Int -> MathMode

fromEnum :: MathMode -> Int

enumFrom :: MathMode -> [MathMode]

enumFromThen :: MathMode -> MathMode -> [MathMode]

enumFromTo :: MathMode -> MathMode -> [MathMode]

enumFromThenTo :: MathMode -> MathMode -> MathMode -> [MathMode]

Eq MathMode

Methods

(==) :: MathMode -> MathMode -> Bool

(/=) :: MathMode -> MathMode -> Bool

Show MathMode

Methods

showsPrec :: Int -> MathMode -> ShowS

show :: MathMode -> String

showList :: [MathMode] -> ShowS

setPointerMode :: Handle -> PointerMode -> IO ()

Set the pointer mode used by cuBLAS library functions. For example, this controls whether the scaling parameters \(\alpha\) and \(\beta\) of the ?gemm operation are treated as residing in host or device memory.

The default mode is for values to be passed by reference from the host.

http://docs.nvidia.com/cuda/cublas/index.html#cublassetpointermode

getPointerMode :: Handle -> IO PointerMode

Get the pointer mode used by cuBLAS library functions to pass scalar arguments.

http://docs.nvidia.com/cuda/cublas/index.html#cublasgetpointermode

setAtomicsMode :: Handle -> AtomicsMode -> IO ()

Set whether cuBLAS library functions are allowed to use atomic functions, when available. The implementations are generally faster, but can generate results which are not strictly identical from one run to another.

http://docs.nvidia.com/cuda/cublas/index.html#cublassetatomicsmode

getAtomicsMode :: Handle -> IO AtomicsMode

Determine whether cuBLAS library functions are allowed to use atomic operations.

http://docs.nvidia.com/cuda/cublas/index.html#cublasgetatomicsmode

setMathMode :: Handle -> MathMode -> IO ()

Set whether cuBLAS library functions are allowed to use Tensor Core operations where available.

http://docs.nvidia.com/cuda/cublas/index.html#cublassetmathmode

since 0.4.0.0

getMathMode :: Handle -> IO MathMode

Determine whether cuBLAS library functions are allowed to use Tensor Core operations where available.

http://docs.nvidia.com/cuda/cublas/index.html#cublasgetmathmode

since 0.4.0.0