From 0b3c0f11bc18538a41ef0e1b649de9d58d07a067 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Fri, 9 Jun 2023 09:35:33 -0700
Subject: [PATCH] Add cu12 packages to allow list for download.pytorch.org

---
 s3_management/manage.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/s3_management/manage.py b/s3_management/manage.py
index 757ac0833..cb5783f38 100644
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -61,6 +61,17 @@
     "nvidia_cusparse_cu11",
     "nvidia_nccl_cu11",
     "nvidia_nvtx_cu11",
+    "nvidia_cublas_cu12",
+    "nvidia_cuda_cupti_cu12",
+    "nvidia_cuda_nvrtc_cu12",
+    "nvidia_cuda_runtime_cu12",
+    "nvidia_cudnn_cu12",
+    "nvidia_cufft_cu12",
+    "nvidia_curand_cu12",
+    "nvidia_cusolver_cu12",
+    "nvidia_cusparse_cu12",
+    "nvidia_nccl_cu12",
+    "nvidia_nvtx_cu12",
     "packaging",
     "portalocker",
     "pytorch_triton",