[Codegen] Fix transfer_scatter failing tests (#24043)

The commit didn't account for canonicalizations kicking in (which were
landed as part of another pr).

compiler/src/iree/compiler/Codegen/LLVMGPU/test/vector_lowering.mlir

diff --git a/compiler/src/iree/compiler/Codegen/LLVMGPU/test/vector_lowering.mlir b/compiler/src/iree/compiler/Codegen/LLVMGPU/test/vector_lowering.mlir
index f8727cc..c602d6b 100644
--- a/compiler/src/iree/compiler/Codegen/LLVMGPU/test/vector_lowering.mlir
+++ b/compiler/src/iree/compiler/Codegen/LLVMGPU/test/vector_lowering.mlir
@@ -319,9 +319,9 @@
   return
 }
 
-// After unrolling, the 2D scatter becomes 4 rank-1 sub-scatters.
+// After unrolling, the 2D scatter becomes 4 rank-1 stores.
 // CHECK-LABEL: func.func @transfer_scatter_unroll_embedding_write
-// CHECK-COUNT-4: transfer_scatter {{.+}} : vector<64xf16>
+// CHECK-COUNT-4: vector.store {{.+}} : memref<4096x64xf16>, vector<64xf16>
 
 // -----
 
@@ -342,9 +342,9 @@
   return
 }
 
-// After unrolling, mask slices are passed to each sub-scatter.
+// After unrolling, mask slices are passed to each masked store.
 // CHECK-LABEL: func.func @transfer_scatter_unroll_masked
-// CHECK-COUNT-4: transfer_scatter {{.+}} : vector<64xf16>
+// CHECK-COUNT-4: vector.maskedstore {{.+}} : memref<4096x64xf16>, vector<64xi1>, vector<64xf16>
 
 // -----
 
@@ -363,10 +363,10 @@
   return %out : tensor<4096x64xf16>
 }
 
-// After unrolling, the 2D scatter becomes 4 rank-1 sub-scatters chained
+// After unrolling, the 2D scatter becomes 4 rank-1 transfer_write chained
 // via tensor SSA results.
 // CHECK-LABEL: func.func @transfer_scatter_unroll_tensor
-// CHECK-COUNT-4: transfer_scatter {{.+}} : vector<64xf16>
+// CHECK-COUNT-4: vector.transfer_write {{.+}} : vector<64xf16>, tensor<4096x64xf16>
 
 // -----
 
@@ -389,6 +389,6 @@
 }
 
 // After two rounds of unrolling (d0=4 then d1=8), the 3D scatter
-// becomes 4*8=32 rank-1 sub-scatters.
+// becomes 4*8=32 rank-1 stores.
 // CHECK-LABEL: func.func @transfer_scatter_unroll_transposed_index
-// CHECK-COUNT-32: transfer_scatter {{.+}} : vector<64xf16>
+// CHECK-COUNT-32: vector.store {{.+}} : memref<4096x64xf16>, vector<64xf16>