Containerize ROCm build and move it to AMD GPU runners (facebookresea…

…rch#3747)

Summary:
Pull Request resolved: facebookresearch#3747

This change converts the ROCm build to run inside containers and updates it to run on AMD GPU based runners. Still working with the AMD team to resolve test failures before enabled those.

Differential Revision: D61049115

fbshipit-source-id: 28274e0bde795f99b3d78711beaf9b3ed3c5e66c

.github/workflows/build.yml

@@ -93,8 +93,19 @@ jobs:
   linux-x86_64-GPU-w-ROCm-cmake:
     name: Linux x86_64 GPU w/ ROCm (cmake)
     needs: linux-x86_64-cmake
-    runs-on: 4-core-ubuntu
+    runs-on: faiss-amd-MI200
+    container:
+      image: ubuntu:22.04
+      options: --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN
     steps:
+      - name: Container setup
+        run: |
+            if [ -f /.dockerenv ]; then
+              apt-get update && apt-get install -y sudo && apt-get install -y git
+              git config --global --add safe.directory '*'
+            else
+              echo 'Skipping. Current job is not running inside a container.'
+            fi
       - name: Checkout
         uses: actions/checkout@v4
       - name: Build and Test (cmake)