function

bench.detect_variant

def detect_variant() -> str

Map the running PyTorch build to a wheel-variant tag.

Returns 'cpu' for CPU-only builds; 'cuXYZ' otherwise (matches the wheel suffixes produced by release.yml).

Source: torchmatch/bench/_filename.py:22