From 5e2c1d28d971d2e055251e9c1b04cba9bedc9de1 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Mon, 23 Sep 2024 17:25:23 -0400 Subject: [PATCH 1/2] Add torchrec dependencies to allow list List: https://github.com/pytorch/torchrec/pull/2273 --- s3_management/manage.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/s3_management/manage.py b/s3_management/manage.py index f6d64b928c..0524500d0a 100755 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -109,6 +109,7 @@ "filelock", "fsspec", "idna", + "iopath", "intel_openmp", "Jinja2", "lit", @@ -143,6 +144,7 @@ "nvidia_nvjitlink_cu12", "packaging", "portalocker", + "pyre-extensions", "pytorch_triton", "pytorch_triton_rocm", "pytorch_triton_xpu", From b3e522deafb0479f766c90a2bb9321fab39bc001 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Mon, 23 Sep 2024 17:31:51 -0400 Subject: [PATCH 2/2] Update s3_management/manage.py --- s3_management/manage.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/s3_management/manage.py b/s3_management/manage.py index 0524500d0a..0dbc6fb067 100755 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -144,7 +144,7 @@ "nvidia_nvjitlink_cu12", "packaging", "portalocker", - "pyre-extensions", + "pyre_extensions", "pytorch_triton", "pytorch_triton_rocm", "pytorch_triton_xpu",