From 1f978fcd9fc5b0ac527fce1907266bdca70a4b21 Mon Sep 17 00:00:00 2001
From: omkar kakarparthi <75638701+okakarpa@users.noreply.github.com>
Date: Sun, 22 Sep 2024 21:35:50 -0500
Subject: [PATCH] Update sanity.yml

---
 .github/workflows/sanity.yml | 34 +++++++++++++++++++++-------------
 1 file changed, 21 insertions(+), 13 deletions(-)

diff --git a/.github/workflows/sanity.yml b/.github/workflows/sanity.yml
index b993dca33978..0cbf88a9902d 100644
--- a/.github/workflows/sanity.yml
+++ b/.github/workflows/sanity.yml
@@ -1,19 +1,27 @@
-name: Print Environment Variables
+name: Run ROCm PyTorch Microbenchmarks on k8
 
 on: [push]
 
 jobs:
-  print-env:
-    runs-on: arc-runner-set-rocm-gpu
+  benchmark:
+    runs-on: [arc-runner-set-rocm-gpu]
+
+    container:
+      image: rocm/pytorch:latest
+      options: >-
+        -it --device=/dev/kfd --device=/dev/dri
+        -u root --ipc=host --shm-size 16G --group-add video
+        --cap-add=SYS_PTRACE --security-opt seccomp=unconfined
     steps:
-      - name: Print Environment Variables
-        run: printenv   
-      - name: Try rocm-smi
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: check rocm is working
+        run: |
+          rocm-smi
+          rocminfo
+      - name: Run Microbenchmarks
         run: |
-          docker run \
-            --device=/dev/kfd --device=/dev/dri \
-            --group-add video \
-            --cap-add=SYS_PTRACE \
-            --security-opt seccomp=unconfined \
-            --shm-size=64G \
-            rocm/rocm-terminal rocm-smi
+          git clone https://github.com/ROCm/pytorch-micro-benchmarking.git
+          cd pytorch-micro-benchmarking
+          python3 micro_benchmarking_pytorch.py --network resnet50