diff --git a/.github/workflows/ait_ci.yml b/.github/workflows/ait_ci.yml
index 8626d301b..b73812749 100644
--- a/.github/workflows/ait_ci.yml
+++ b/.github/workflows/ait_ci.yml
@@ -2,11 +2,11 @@ name: AITemplate_ci
 
 on:
   pull_request:
-     tags:
-       - rocm_ci*
+     types: [labeled]
 
 jobs:
   build:
+    if: contains(github.event.label.name, 'rocm')
     runs-on: rocm
 
     steps: