pytorch
de86538f - [ROCM] Restrict pytorch rocm to only use triton 2.0.x (#95793)

Commit
2 years ago
[ROCM] Restrict pytorch rocm to only use triton 2.0.x (#95793) To align with upstream, we are requiring triton dependency to be between 2.0.0 and 2.1. This will allow PyTorch 2.0 on ROCM to stay flexible enough to pick up any performance/stability improvements from Triton, without needing to cut a separate PyTorch version. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95793 Approved by: https://github.com/huydhn
Author
Committer
Parents
Loading