编译#

class nvmath.device.random.Compile( cc: ComputeCapability | None = None, )[source]#

编译具有指定计算能力的随机设备 API。

应将 files 和 extension 属性用作 numba.cuda.jit() 装饰器的参数，在 Numba 内核中使用随机设备 API。

参数:: cc – (可选) 计算能力，指定为 nvmath.device.ComputeCapability 类型的对象。如果未指定，将使用默认计算能力。

示例

>>> from numba import cuda
>>> from nvmath.device import random
>>> compiled_apis = random.Compile()
>>> @cuda.jit(link=compiled_apis.files, extensions=compiled_apis.extension)
... def kernel():
...     pass  # use random device APIs here

方法

__init__( cc: ComputeCapability | None = None, )[source]#

属性

extension#: 使用 Numba 内核的随机设备 API 所需的扩展。

files#: 将随机设备 API 与 Numba 内核链接所需的数据。