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

Commit
2 years ago
[ROCM] Restrict pytorch rocm to only use triton 2.0.x (#95793) (#95834) 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
Parents
Loading