diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index 82f536582..0727bf420 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -4,13 +4,13 @@ on:
   workflow_dispatch:
   pull_request:
     branches:
-      - main
+      - triton-mlir
 
 jobs:
 
   Integration-Tests:
 
-    runs-on: self-hosted
+    runs-on: [self-hosted, A10]
 
     steps: