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.
currently supports cuBLASDx 0.1.1, also available
as part of MathDx 24.04.
API Reference#
|
Create an |
|
A class that encapsulates a partial BLAS device function. |
|
A namedtuple class that encapsulates the three leading dimensions in matrix multiplication \(C = \alpha Op(A) Op(B) + \beta C\). |
|
A namedtuple class that encapsulates the transpose mode for input matrices |