Add basic Rocm CI build and test

.github/pytorch-probot.yml

@@ -0,0 +1,2 @@
+ciflow_push_tags:
+- ciflow/rocm

.github/workflows/linux_rocm.yml

@@ -0,0 +1,140 @@
+name: Linux ROCm build and test
+
+on:
+  push:
+    tags:
+      - ciflow/rocm/*
+    branches:
+      - main
+
+env:
+  DOCKER_IMAGE: "pytorch/manylinux2_28-builder:rocm7.2"
+
+permissions:
+  id-token: write
+  contents: read
+
+jobs:
+  pr-test:
+    runs-on: linux.rocm.gpu.gfx942.1
+    timeout-minutes: 180 # 3 hours
+    steps:
+      - name: Checkout Kineto
+        uses: actions/checkout@v5
+        with:
+          path: kineto
+          submodules: recursive
+
+      - name: Pull docker image
+        uses: pytorch/test-infra/.github/actions/pull-docker-image@main
+        with:
+          docker-image: ${{ env.DOCKER_IMAGE }}
+
+      - name: Setup ROCM
+        uses: pytorch/pytorch/.github/actions/setup-rocm@main
+
+      - name: Get env vars
+        run: |
+          echo GITHUB_WORKFLOW   = $GITHUB_WORKFLOW
+          echo HOME              = $HOME
+          echo GITHUB_ACTION     = $GITHUB_ACTION
+          echo GITHUB_ACTIONS    = $GITHUB_ACTIONS
+          echo GITHUB_REPOSITORY = $GITHUB_REPOSITORY
+          echo GITHUB_EVENT_NAME = $GITHUB_EVENT_NAME
+          echo GITHUB_EVENT_PATH = $GITHUB_EVENT_PATH
+          echo GITHUB_WORKSPACE  = $GITHUB_WORKSPACE
+          echo GITHUB_SHA        = $GITHUB_SHA
+          echo GITHUB_REF        = $GITHUB_REF
+
+      - name: Build libkineto (static and shared library)
+        run: |
+          container_name=$(docker run \
+            ${GPU_FLAG:-} \
+            --tty \
+            --detach \
+            --shm-size=32gb \
+            -v "${PWD}/kineto:/kineto" \
+            -w / \
+            "${{ env.DOCKER_IMAGE }}"
+          )
+          echo "Container name: ${container_name}"
+
+          docker exec -t -w "/kineto" "${container_name}" bash -c "
+            set -eux
+            mkdir build_static
+            mkdir build_shared
+            pip install cmake
+          "
+
+          docker exec -t -w "/kineto/build_static" "${container_name}" bash -c "
+            set -eux
+            cmake -DKINETO_LIBRARY_TYPE=static ../libkineto/
+            make -j
+          "
+
+          docker exec -t -w "/kineto/build_shared" "${container_name}" bash -c "
+            set -eux
+            cmake -DKINETO_LIBRARY_TYPE=shared ../libkineto/
+            make -j
+          "
+
+      - name:  Run libkineto tests
+        run: |
+          container_name=$(docker ps -lq)
+          docker exec -t -w "/kineto/build_static" "${container_name}" bash -c "make test"
+
+      - name: Clone PyTorch
+        run: |
+          container_name=$(docker ps -lq)
+          docker exec -t -w "/" "${container_name}" bash -c "
+            set -eux
+            git clone --recursive --branch viable/strict https://github.com/pytorch/pytorch.git
+          "
+
+      - name: Replace PyTorch's Kineto with PR version
+        run: |
+          container_name=$(docker ps -lq)
+          docker exec -t -w "/pytorch" "${container_name}" bash -c "
+            set -eux
+            rm -rf third_party/kineto
+            ln -s /kineto third_party/kineto
+          "
+
+      - name: Build PyTorch from source
+        run: |
+          container_name=$(docker ps -lq)
+          docker exec -t -w "/pytorch" "${container_name}" bash -c "
+            set -eux
+            pip install -r requirements.txt
+            export USE_ROCM=1
+            export BUILD_TEST=1
+            python setup.py develop
+          "
+
+      - name: Run PyTorch profiler tests
+        run: |
+          container_name=$(docker ps -lq)
+          docker exec -t -w "/pytorch" "${container_name}" bash -c "
+            set -eux
+            # TODO: Dynamically add/remove tests to the exclusion list based on their
+            # status on trunk instead of maintaining a hardcoded list of known failures.
+            # This will prevent the list from becoming stale as tests get fixed upstream.
+            python -m pytest test/profiler/ -v \
+              --deselect=test/profiler/test_memory_profiler.py::TestDataFlow::test_data_flow_graph_complicated \
+              --deselect=test/profiler/test_memory_profiler.py::TestMemoryProfilerE2E::test_categories_e2e_sequential_fwd_bwd \
+              --deselect=test/profiler/test_memory_profiler.py::TestMemoryProfilerE2E::test_categories_e2e_simple_fwd_bwd \
+              --deselect=test/profiler/test_memory_profiler.py::TestMemoryProfilerE2E::test_categories_e2e_simple_fwd_bwd_step \
+              --deselect=test/profiler/test_profiler.py::TestProfiler::test_kineto \
+              --deselect=test/profiler/test_profiler.py::TestProfiler::test_user_annotation \
+              --deselect=test/profiler/test_profiler.py::TestExperimentalUtils::test_fuzz_symbolize \
+              --deselect=test/profiler/test_profiler.py::TestExperimentalUtils::test_profiler_debug_autotuner \
+              --deselect=test/profiler/test_torch_tidy.py::TestTorchTidyProfiler::test_tensorimpl_invalidation_scalar_args
+          "
+
+      - name: Teardown Linux
+        uses: pytorch/test-infra/.github/actions/teardown-linux@main
+        if: always()
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
+  cancel-in-progress: true