cuBLASDx API (nvmath.device)

概述

这些 API 提供了与 NVIDIA cuBLASDx 库的集成。cuBLASDx 的详细文档可以在 cuBLASDx 文档中找到。

注意

模块 nvmath.device 中的 matmul 设备 API 当前支持 cuBLASDx 0.1.1，它也包含在 MathDx 24.04 中。

API 参考

matmul(*[, compiler])	创建一个 BlasOptions 对象，该对象封装了一个已编译且可直接使用的用于矩阵乘法的设备函数。
BlasOptions(size, precision, data_type, *[, ...])	一个封装部分 BLAS 设备函数的类。
LeadingDimension(a, b, c)	一个 namedtuple 类，封装了矩阵乘法 \(C = \alpha Op(A) Op(B) + \beta C\) 中的三个主导维度。
TransposeMode(a, b)	一个 namedtuple 类，封装了矩阵乘法中输入矩阵 A 和 B 的转置模式。