FFT Operators¶

本页暂无中文版。以下为英文原文。

Every op on this page is used the same way: construct it once, then call it. The constructor takes what the kernel is compiled with; the call takes the tensors. Both are documented under each op — __init__ and forward, where forward is what runs when you call op(...).

Complex-to-complex¶

tileops.ops.fft.FFTC2CFwdOp ¶

1D Complex-to-Complex Fast Fourier Transform operation.

Computes the one-dimensional discrete Fourier transform of complex input. This is equivalent to torch.fft.fft.

Supports batched input: any leading dimensions are flattened into a single batch dimension, processed in parallel by the kernel, and reshaped back.

Uses pre-computed twiddle factor LUT and shared-memory butterfly fusion for optimal GPU performance.

init ¶

__init__(
    tune=False,
    kernel_map=None,
)

Build the op. Shapes and dtype are taken from the first call.

Parameters:

tune (bool, default: False ) –

Whether to enable autotuning (default: False)
kernel_map (Optional[Dict[str, Kernel]], default: None ) –

Optional custom kernel mapping for testing

forward ¶

forward(
    input,
)

Compute 1D FFT of complex input.