Skip to contents

Performs matrix cross-product using a GPU. This function is merely a couple of wrappers for the CUBLAS cublasSgemm function.

Usage

gpuCrossprod(a, b=NULL)

Arguments

a

a matrix of floating point values.

b

a matrix of floating point values. A null value defaults to 'a'.

Value

A matrix of single precision floating point values. The matrix is the cross-product of arguments 'a' and 'b', i.e., t(a) * b.

Examples

if (FALSE) { # gpuAvailable()
matA <- matrix(runif(3*2), 3, 2)
matB <- matrix(runif(3*4), 3, 4)
gpuCrossprod(matA, matB)
}