Add nvidia_nvshmem_cu12 to list of allowed files (#6699)

atalman · web-flow · commit 6105c6f94a60 · 2025-06-03T12:11:28.000-04:00
To unblock pytorch/pytorch#154568
diff --git a/s3_management/manage.py b/s3_management/manage.py
@@ -163,6 +163,7 @@
     "nvidia_cudnn_cu12",
     "nvidia_cufft_cu12",
     "nvidia_cufile_cu12",
+    "nvidia_nvshmem_cu12",
     "nvidia_curand_cu12",
     "nvidia_cusolver_cu12",
     "nvidia_cusparse_cu12",
