From af3b8b5dfc00d7396b8e78f8153d8aaef707b61d Mon Sep 17 00:00:00 2001
From: Wonhyuk Yang <wonhyuk@postech.ac.kr>
Date: Fri, 22 May 2026 21:09:05 +0900
Subject: [PATCH] [CI] Register 5 previously-uncovered tests in
 pytorchsim_test.yml

Adds three top-level jobs (test_eager, test_exponent, test_sort) and two
steps inside the test_fusion job (test_attention_fusion, test_matmul_vector).
All five were verified locally before registration.

Files in tests/ that remain intentionally out of CI:
- test_gqa.py, test_gqa_decode.py: WIP GQA SDPA path, tracked by issue #198
- test_sdpa.py: overlaps with in-flight SDPA template work; ambiguous about
  which backend it actually exercises
- test_topk.py: sort-family coverage now provided by test_sort (stable +
  unstable + duplicate-key); revisit if topk-specific shapes need gating
- test_group_conv.py: not run locally yet (stress config); follow-up after
  runtime cost is understood
- test_vectorops.py: imports from other tests (test_add, test_activation,
  test_reduce, test_layernorm, test_softmax) which is fragile; needs an
  independent helper extraction before going into CI

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
---
 .github/workflows/pytorchsim_test.yml | 73 +++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)

diff --git a/.github/workflows/pytorchsim_test.yml b/.github/workflows/pytorchsim_test.yml
index 4b4fab80..e7ae6385 100644
--- a/.github/workflows/pytorchsim_test.yml
+++ b/.github/workflows/pytorchsim_test.yml
@@ -517,6 +517,22 @@ jobs:
             -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
             ${{ inputs.image_name }} python3 PyTorchSim/tests/Fusion/test_conv_fusion.py
 
+      - name: Run test_attention_fusion.py
+        run: |
+          echo "Running test_attention_fusion.py"
+          docker run --rm \
+            -e vpu_num_lanes="${{ inputs.vector_lane }}" \
+            -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} python3 PyTorchSim/tests/Fusion/test_attention_fusion.py
+
+      - name: Run test_matmul_vector.py
+        run: |
+          echo "Running test_matmul_vector.py"
+          docker run --rm \
+            -e vpu_num_lanes="${{ inputs.vector_lane }}" \
+            -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} python3 PyTorchSim/tests/Fusion/test_matmul_vector.py
+
   test_moe:
     name: Run test_moe
     runs-on: self-hosted
@@ -688,6 +704,63 @@ jobs:
             -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
             ${{ inputs.image_name }} python3 PyTorchSim/tests/DeepSeek/test_deepseek_v3_base.py
 
+  test_eager:
+    name: Run test_eager.py
+    runs-on: self-hosted
+    steps:
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Run test_eager.py
+        run: |
+          echo "Running test_eager.py"
+          docker run --rm \
+            -e vpu_num_lanes="${{ inputs.vector_lane }}" \
+            -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} python3 PyTorchSim/tests/test_eager.py
+
+  test_exponent:
+    name: Run test_exponent.py
+    runs-on: self-hosted
+    steps:
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Run test_exponent.py
+        run: |
+          echo "Running test_exponent.py"
+          docker run --rm \
+            -e vpu_num_lanes="${{ inputs.vector_lane }}" \
+            -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} python3 PyTorchSim/tests/test_exponent.py
+
+  test_sort:
+    name: Run test_sort.py
+    runs-on: self-hosted
+    steps:
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Run test_sort.py
+        run: |
+          echo "Running test_sort.py"
+          docker run --rm \
+            -e vpu_num_lanes="${{ inputs.vector_lane }}" \
+            -e vpu_spad_size_kb_per_lane="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} python3 PyTorchSim/tests/test_sort.py
+
   test_accuracy:
     name: Run test_accuracy and test_speedup
     runs-on: self-hosted