DeviceCallable

class nvmath.fft.DeviceCallable(

ltoir: int | bytes | None = None,

size: int | None = None,

data">: int | None = None,

)

一个捕获 LTO-IR 可调用对象的数据类。

ltoir

LTO-IR 格式的设备可调用函数，可以作为 bytes 对象或作为指向 LTO-IR 的 Python int 指针提供。

类型:

int | bytes | None

size

LTO-IR 可调用对象的大小。如果未指定，并且为 bytes 对象传递了 ltoir，则会从中计算大小。如果为 LTO-IR 提供了指针，则必须指定 size。

类型:

int | None

data

指向回调中使用的用户数据的设备指针。 默认为 None，表示回调中将使用空指针。

类型:

int | None

另请参阅

FFT.plan(), fft(), ifft(), rfft(), 和 irfft()。