`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.