设备 API 实用程序 (nvmath.device)#

概述#

nvmath-python 提供了以下设备端实用程序。

API 参考#

current_device_lto()

一个辅助函数，用于获取当前设备上链接时优化 (LTO) 的默认代码类型。

float16x2(x, y)

创建用于 float16 且向量长度为 2 的 Numba 兼容向量对象。

float16x4(x, y, z, w)

创建用于 float16 且向量长度为 4 的 Numba 兼容向量对象。

float32x2(x, y)

创建用于 float32 且向量长度为 2 的 Numba 兼容向量对象。

float64x2(x, y)

创建用于 float64 且向量长度为 2 的 Numba 兼容向量对象。

float16x2_type

用于 float16 且向量长度为 2 的 Numba 兼容向量类型对象

float16x4_type

用于 float16 且向量长度为 4 的 Numba 兼容向量类型对象

float32x2_type

用于 float32 且向量长度为 2 的 Numba 兼容向量类型对象

float64x2_type

用于 float64 且向量长度为 2 的 Numba 兼容向量类型对象

ISAVersion(major, minor)

一个命名元组类，封装了代码版本。

Code(code_type, isa_version, data)

一个命名元组类，封装了代码类型、版本和缓冲区。

CodeType(kind, cc)

一个命名元组类，封装了代码种类和计算能力。

ComputeCapability(major, minor)

一个命名元组类，封装了主版本号和次版本号计算能力。

CodeType(kind, cc)

一个命名元组类，封装了代码种类和计算能力。

Symbol(variant, name)

一个命名元组类，封装了设备函数符号及其映射到的 API。

Dim3([x, y, z])

一个命名元组类，封装了网格和块的维度。
在本页