1// RUN: mlir-opt %s -tensor-bufferize | FileCheck %s
2
3// CHECK-LABEL:   func @extract(
4// CHECK-SAME:                  %[[TENSOR:.*]]: tensor<?xf32>,
5// CHECK-SAME:                  %[[IDX:.*]]: index) -> f32 {
6// CHECK:           %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<?xf32>
7// CHECK:           %[[RET:.*]] = load %[[MEMREF]][%[[IDX]]] : memref<?xf32>
8// CHECK:           return %[[RET]] : f32
9// CHECK:         }
10func @extract(%arg0: tensor<?xf32>, %arg1: index) -> f32 {
11  %0 = tensor.extract %arg0[%arg1] : tensor<?xf32>
12  return %0 : f32
13}
14