diff --git a/requirements/rocm-build.txt b/requirements/rocm-build.txt index 29d56478..05de4ff1 100644 --- a/requirements/rocm-build.txt +++ b/requirements/rocm-build.txt @@ -6,6 +6,7 @@ torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 +triton==3.2 cmake>=3.26,<4 packaging setuptools>=61