chore: add GPUs to docgen

Author: narendasan

.github/workflows/docgen.yml

@@ -13,6 +13,7 @@ jobs:
     runs-on: linux.g5.4xlarge.nvidia.gpu
     container:
       image: docker.io/pytorch/manylinux-builder:cuda12.1
+      options: --gpus all
     env:
       CUDA_HOME: /usr/local/cuda-12.1
       VERSION_SUFFIX: cu121