frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Convenience script for getting correct Triton nightly binary (#96669)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96669
Approved by: https://github.com/ngimel, https://github.com/malfet
This commit is contained in:
Edward Z. Yang
2023-03-13 14:14:00 -04:00
committed by PyTorch MergeBot
parent 120c6f6637
commit a8d1eb1961
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/install_triton_wheel.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/bash
# Updates Triton to the pinned version for this copy of PyTorch
pip install --extra-index-url https://download.pytorch.org/whl/nightly/ "pytorch-triton==$(cat .ci/docker/triton_version.txt)+$(head -c 10 .ci/docker/ci_commit_pins/triton.txt)"