use optimization_barrier in tests

Signed-off-by: Nirvedh Meshram <[email protected]>

Author: nirvedhmeshram

compiler/src/iree/compiler/Codegen/Common/GPU/test/gpu_pack_to_instrinsics.mlir

@@ -312,8 +312,8 @@ func.func @hoist_pack_unpack_multiple_loop(%arg0 : tensor<1x1x4x2x16x16xbf16>, %
 
 // -----
 
-func.func @propagate_extract_basic(%arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
-  %empty = tensor.empty() : tensor<128xf32>
+func.func @propagate_extract_basic(%input : tensor<128xf32>, %arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
+  %empty = util.optimization_barrier %input : tensor<128xf32>
   %extracted_slice =  tensor.extract_slice %empty[%arg0] [%arg0] [1] : tensor<128xf32> to tensor<?xf32>
   %generic = linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> (d0)>], iterator_types = ["parallel"]} ins(%extracted_slice : tensor<?xf32>) outs(%arg1 : tensor<?xbf16>) {
   ^bb0(%in: f32, %out: bf16):
@@ -346,8 +346,8 @@ func.func @no_propagate_extract_blockargument(%input : tensor<128xf32>, %arg0 :
 
 // -----
 
-func.func @no_propagate_extract_differentblock_1(%arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
-  %empty = tensor.empty() : tensor<128xf32>
+func.func @no_propagate_extract_differentblock_1(%input : tensor<128xf32>, %arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
+  %empty = util.optimization_barrier %input : tensor<128xf32>
   %c0 = arith.constant 0 : index
   %c32 = arith.constant 32 : index
   %for = scf.for %arg2 = %c0 to %c32 step %arg0 iter_args(%arg4 = %arg1) -> tensor<?xbf16> {
@@ -368,8 +368,8 @@ func.func @no_propagate_extract_differentblock_1(%arg0 : index, %arg1 : tensor<?
 
 // -----
 
-func.func @no_propagate_extract_differentblock_2(%arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
-  %empty = tensor.empty() : tensor<128xf32>
+func.func @no_propagate_extract_differentblock_2(%input : tensor<128xf32>, %arg0 : index, %arg1 : tensor<?xbf16>) -> tensor<?xbf16> {
+  %empty = util.optimization_barrier %input : tensor<128xf32>
   %c0 = arith.constant 0 : index
   %c32 = arith.constant 32 : index
   %extracted_slice =  tensor.extract_slice %empty[%arg0] [%arg0] [1] : tensor<128xf32> to tensor<?xf32>