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

dllehr-amd · web-flow · commit c04134cdb136 · 2023-03-01T19:10:00.000-05:00
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: #95793 Approved by: https://github.com/huydhn
diff --git a/.github/requirements/triton-requirements-rocm.txt b/.github/requirements/triton-requirements-rocm.txt
@@ -1 +1 @@
-pytorch-triton-rocm>=2.0.0.dev
+pytorch-triton-rocm>=2.0.0,<2.1

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-pytorch-triton-rocm>=2.0.0.dev`
	`1`	`+pytorch-triton-rocm>=2.0.0,<2.1`