How to know which triton version to build for a given pytorch version

1

I am trying to make the inductor backend of torchdynamo work on Jetson AGX Orin (aarch64 iGPU system). I have to use torch version 2.3, and I compiled triton v2.1 and also the main branch from the source and installed it. However, both have compatibility issues, resulting in errors like no such package found triton.common etc. How can I know which branch or commit of triton I should compile to make it work with torch 2.3 (or any other version)?

2

You can find the Triton commit in .ci/docker/ci_commit_pins/triton.txt.

2 Likes
3

I was having trouble checking out that commit. I realized that I needed to fetch it like this:
git fetch origin 79c6c9b209a5692b9a895398f4f3a033f8f80415
git checkout 79c6c9b209a5692b9a895398f4f3a033f8f80415

Thanks.