[CI][Proton][AMD] Fix test_cmd and enable CI on AMD (triton-lang#4026)

Co-authored-by: Keren Zhou <kerenzhou@openai.com>
Co-authored-by: Jokeren <robinho364@gmail.com>

.github/workflows/integration-tests.yml

@@ -399,6 +399,10 @@ jobs:
 
           # Run runtime tests serially to avoid race condition with cache handling
           python3 -m pytest -s runtime
+      - name: Run Proton tests
+        run: |
+          cd third_party/proton
+          python3 -m pytest test
       - name: Run C++ unittests
         run: |
           cd python

.github/workflows/integration-tests.yml.in

@@ -404,6 +404,11 @@ jobs:
           # Run runtime tests serially to avoid race condition with cache handling
           python3 -m pytest -s runtime
 
+      - name: Run Proton tests
+        run: |
+          cd third_party/proton
+          python3 -m pytest test
+
       - *run-cpp-unittests-step
       - *save-build-artifacts-step
       - *inspect-cache-directories-step

third_party/proton/test/test_cmd.py

@@ -27,4 +27,4 @@ def test_exec(mode):
         data = json.load(f, )
         kernels = data[0]["children"]
         assert len(kernels) == 2
-        assert kernels[1]["frame"]["name"] == "test"
+        assert kernels[0]["frame"]["name"] == "test" or kernels[1]["frame"]["name"] == "test"