diff --git a/s3_management/manage.py b/s3_management/manage.py
index 3ce72f646..e929b0680 100644
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -47,6 +47,7 @@
     "numpy",
     "packaging",
     "pytorch_triton",
+    "pytorch_triton_rocm",
     "requests",
     "sympy",
     "torch",