Fix lit test after rebase

Signed-off-by: Harsh Menon <harsh@nod-labs.com>
iree-org · Jan 29, 2025 · fe4c42d · fe4c42d
1 parent 27b0d3d
commit fe4c42d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/lit_tests/kernel/wave/attention.py b/lit_tests/kernel/wave/attention.py
@@ -1171,13 +1171,12 @@ def test_prefill_attention():
         # CHECK-COUNT-1:            vector.store
         # CHECK-COUNT-1:            vector.maskedload
         # CHECK-COUNT-1:            vector.store
-        # CHECK-COUNT-16:           vector.load
-        # CHECK-COUNT-16:           amdgpu.mfma
-        # CHECK-COUNT-4:            gpu.shuffle xor {{.*}}
         # CHECK-COUNT-1:            vector.maskedload
         # CHECK-COUNT-1:            vector.store
         # CHECK-COUNT-1:            vector.maskedload
         # CHECK-COUNT-1:            vector.store
-        # CHECK-COUNT-16:           vector.load
+        # CHECK-COUNT-32:           vector.load
+        # CHECK-COUNT-16:           amdgpu.mfma
+        # CHECK-COUNT-4:            gpu.shuffle xor {{.*}}
         # CHECK-COUNT-16:           amdgpu.mfma
         # CHECK-COUNT-16:      vector.maskedstore