fused_fftconv2d_arch_supported

fused_fftconv2d_arch_supported(device=None, fft_size=128)

Whether device can run the fused 2D kernel at this FFT tile size.

Tiles below 128 run on any CUDA device the kernel supports; the 128 tile additionally needs SM90+ (see FUSED_FFT_SIZE_128_MIN_ARCH).

Parameters:

• device (device | None) – CUDA device to probe. Defaults to the current device.

• fft_size (int) – Tile size to check, as returned by resolve_fused_fft_size().

Returns:

True if the device can run that tile, False otherwise (including on CPU or when CUDA is unavailable).

Return type:

bool