`hilbertsfc.torch.types`¶

hilbertsfc.torch.types ¶

CPUBackend = Literal['auto', 'numba', 'torch']

CPU backend options for hilbertsfc.torch functions.

GPUBackend = Literal['auto', 'triton', 'torch']

GPU backend options for hilbertsfc.torch functions.

TorchCacheMode = Literal['device', 'host_only']

Cache mode for torch look-up tables (LUTs).

TorchDeviceLike = device | str | None

Device specifier with the same semantics as torch.device.

TorchHilbertOp = Literal['hilbert_encode_2d', 'hilbert_decode_2d', 'hilbert_encode_3d', 'hilbert_decode_3d', 'all']

Identifiers for Hilbert operations, including "all" option to target all operations.

TritonTuningMode = Literal['heuristic', 'autotune_bucketed', 'autotune_exact']

Triton launch tuning options for hilbertsfc.torch functions.