From ec3c748fa2b8c2c01ecf6b5a4efa4ab35d805488 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 14 Sep 2023 23:48:45 +0000 Subject: [PATCH] Document Triton dependency for the release process (#109296) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Document triton dependency for the Pytorch Release ### 🤖 Generated by Copilot at e9773f0 Add documentation for Triton dependency in `RELEASE.md`. The documentation covers how to install and use Triton for various PyTorch builds and platforms. Pull Request resolved: https://github.com/pytorch/pytorch/pull/109296 Approved by: https://github.com/malfet --- RELEASE.md | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/RELEASE.md b/RELEASE.md index 581218c4f97..bc19c8d49f6 100644 --- a/RELEASE.md +++ b/RELEASE.md @@ -35,6 +35,7 @@ - [Submitting Tutorials](#submitting-tutorials) - [Special Topics](#special-topics) - [Updating submodules for a release](#updating-submodules-for-a-release) + - [Triton dependency for the release](#triton-dependency-for-the-release) @@ -352,3 +353,19 @@ git config --file=.gitmodules -e An example of this process can be found here: * https://github.com/pytorch/pytorch/pull/48312 + +## Triton dependency for the release + +In nightly builds for conda and wheels pytorch depend on Triton build by this workflow: https://hud.pytorch.org/hud/pytorch/pytorch/nightly/1?per_page=50&name_filter=Build%20Triton%20Wheel. The pinned version of triton used by this workflow is specified here: https://github.com/pytorch/pytorch/blob/main/.ci/docker/ci_commit_pins/triton.txt . + +In Nightly builds we have following configuration: +* Conda builds, depend on: https://anaconda.org/pytorch-nightly/torchtriton +* Wheel builds, depend on : https://download.pytorch.org/whl/nightly/pytorch-triton/ +* Rocm wheel builds, depend on : https://download.pytorch.org/whl/nightly/pytorch-triton-rocm/ + +However for release we have following : +* Conda builds, depend on: https://anaconda.org/pytorch-test/torchtriton for test and https://anaconda.org/pytorch/torchtriton for release +* Wheel builds, depend only triton pypi package: https://pypi.org/project/triton/ for both test and release +* Rocm wheel builds, depend on : https://download.pytorch.org/whl/test/pytorch-triton-rocm/ for test and https://download.pytorch.org/whl/pytorch-triton-rocm/ for release + +Important: The release of https://pypi.org/project/triton/ needs to be requested from OpenAI once branch cut is completed. Please include the release PIN hash in the request: https://github.com/pytorch/pytorch/blob/release/2.1/.ci/docker/ci_commit_pins/triton.txt .