cuBLASDx APIs (nvmath.device)#

Overview#

These APIs offer integration with the NVIDIA cuBLASDx library. Detailed documentation of cuBLASDx can be found in the cuBLASDx documentation.

Note

The matmul device API in module nvmath.device currently supports cuBLASDx 0.1.1, also available as part of MathDx 24.04.

API Reference#

matmul(*[, compiler])

Create an BlasOptions object that encapsulates a compiled and ready-to-use device function for matrix multiplication.

BlasOptions(size, precision, data_type, *[, ...])

A class that encapsulates a partial BLAS device function.

LeadingDimension(a, b, c)

A namedtuple class that encapsulates the three leading dimensions in matrix multiplication \(C = \alpha Op(A) Op(B) + \beta C\).

TransposeMode(a, b)

A namedtuple class that encapsulates the transpose mode for input matrices A and B in matrix multiplication.