pmeier / light-the-torch

Install PyTorch distributions with computation backend auto-detection
BSD 3-Clause "New" or "Revised" License
223 stars 7 forks source link

Patched `PYTORCH_DISTRIBUTIONS` is no longer correct #97

Closed github-actions[bot] closed 2 years ago

github-actions[bot] commented 2 years ago

The variable light_the_torch._patch.PYTORCH_DISTRIBUTIONS is no longer aligned with the wheels hosted by PyTorch. Please replace it with

PYTORCH_DISTRIBUTIONS = {
    "pytorch_triton",
    "torch",
    "torch_model_archiver",
    "torch_tb_profiler",
    "torcharrow",
    "torchaudio",
    "torchcsprng",
    "torchdata",
    "torchdistx",
    "torchserve",
    "torchtext",
    "torchtriton",
    "torchvision",
}
pmeier commented 2 years ago

If I'm not mistaken, as of nowpytorch_triton only got a single Linux wheel for five Python versions: https://download.pytorch.org/whl/nightly/pytorch_triton/

Screenshot from 2022-10-23 22-18-40

As such, I think we can safely ignore it for now.

pmeier commented 2 years ago

Same goes for torchtriton: https://download.pytorch.org/whl/nightly/torchtriton/

Screenshot from 2022-10-23 22-30-12