[ci]Add attention shapes in boo (#2893)

enable attention in boo.

Co-authored-by: Keshav Vinayak Jha <keshavvinayakjha@gmail.com>

Author: yash-amd

.github/workflows/ci_boo_on_mi355.yml

@@ -13,6 +13,7 @@
 # 4. batch norm with untuned boo and native pytorch
 # 5. prod(nchw) with untuned boo and native pytorch
 # 6. proxy(nchw) with untuned boo and native pytorch
+# 7. Attention shapes with untuned boo and inductor
 
 name: CI - Boo on Mi355
 
@@ -59,6 +60,11 @@ on:
       #   description: "Run All Proxy NCHW"
       #   type: boolean
       #   default: false
+      run_attention:
+        description: "Run Attention"
+        type: boolean
+        default: false
+
   schedule:
     - cron: "45 11 * * *"
 
@@ -239,6 +245,16 @@ jobs:
             --backend=torch --backend=iree_boo_experimental --verify-numerics --numerics-reference-dtype=float32 --numerics-min-samples 100 --verbose --numerics-verbose \
             2>&1 | tee output_artifacts/all_proxy_miopen_iree_nchw.log || true
 
+    - name: Run Attention
+      if: always() && (github.event_name == 'schedule' || inputs.run_attention == true)
+      run: |
+          echo "============ Running attention shapes with Inductor and IREE without tuning ==================="
+          pip install -r amdsharktuner/requirements-dev.txt
+          iree-boo-driver \
+            --commands-file amd-shark-ai-reports/boo/attention_shapes.txt --csv output_artifacts/attention_shapes_miopen_iree.csv \
+            --backend inductor --backend=iree_boo_experimental \
+            2>&1 | tee output_artifacts/attention_shapes.log || true
+
     - name: Run hipblaslt-bench
       if: always() && (github.event_name == 'schedule' || inputs.run_hipblaslt == true)
       run: |

.github/workflows/ci_boo_on_rdna4.yml

@@ -10,6 +10,7 @@
 # 1. proxy with untuned boo and native pytorch
 # 2. prod with untuned boo and native pytorch
 # 3. gemms with untuned boo and native pytorch
+# 4. attention shapes with untuned boo and inductor
 
 name: CI - Boo on RDNA4
 
@@ -45,6 +46,11 @@ on:
         description: "Run Batch Norm"
         type: boolean
         default: false
+      run_attention:
+        description: "Run Attention"
+        type: boolean
+        default: false
+
   schedule:
     - cron: "36 5 * * *"
 
@@ -196,6 +202,16 @@ jobs:
             --backend=torch --backend=iree_boo_experimental --verify-numerics --numerics-reference-dtype=float32 --numerics-min-samples 100 --verbose --numerics-verbose \
             2>&1 | tee output_artifacts/rdna4_all_proxy_miopen_iree_nchw.log || true
 
+    - name: Run Attention
+      if: always() && (github.event_name == 'schedule' || inputs.run_attention == true)
+      run: |
+       echo "============ Running attention shapes with Inductor and IREE without tuning ==================="
+       pip install -r amdsharktuner/requirements-dev.txt
+       iree-boo-driver \
+            --commands-file amd-shark-ai-reports/boo/attention_shapes.txt --csv output_artifacts/rdna4_attention_shapes_miopen_iree.csv \
+            --backend inductor --backend=iree_boo_experimental \
+            2>&1 | tee output_artifacts/rdna4_attention_shapes.log || true
+
     - name: Upload log files
       if: always()
       uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f