1// RUN: mlir-opt %s | mlir-opt | FileCheck %s 2// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s 3 4#CSR = #sparse_tensor.encoding<{ 5 map = (d0, d1) -> (d0 : dense, d1 : compressed) 6}> 7 8// CHECK-LABEL: func @test_clone 9func.func @test_clone(%buf : memref<*xf32>) -> memref<*xf32> { 10 %clone = bufferization.clone %buf : memref<*xf32> to memref<*xf32> 11 return %clone : memref<*xf32> 12} 13 14// CHECK-LABEL: test_to_memref 15func.func @test_to_memref(%arg0: tensor<?xi64>, %arg1: tensor<*xi64>) 16 -> (memref<?xi64, affine_map<(d0) -> (d0 + 7)>>, memref<*xi64, 1>) { 17 %0 = bufferization.to_memref %arg0 18 : tensor<?xi64> to memref<?xi64, affine_map<(d0) -> (d0 + 7)>> 19 %1 = bufferization.to_memref %arg1 20 : tensor<*xi64> to memref<*xi64, 1> 21 return %0, %1 : memref<?xi64, affine_map<(d0) -> (d0 + 7)>>, memref<*xi64, 1> 22} 23 24// CHECK-LABEL: func @test_to_tensor 25func.func @test_to_tensor(%buf : memref<2xf32>) -> tensor<2xf32> { 26 %tensor = bufferization.to_tensor %buf restrict writable : memref<2xf32> to tensor<2xf32> 27 return %tensor : tensor<2xf32> 28} 29 30// CHECK-LABEL: func @test_alloc_tensor_op 31func.func @test_alloc_tensor_op(%t: tensor<?x5xf32>, %sz: index) 32 -> tensor<?x5xf32> 33{ 34 // CHECK: bufferization.alloc_tensor(%{{.*}}) : tensor<?x5xf32> 35 %0 = bufferization.alloc_tensor(%sz) : tensor<?x5xf32> 36 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) : tensor<?x5xf32> 37 %1 = bufferization.alloc_tensor() copy(%t) : tensor<?x5xf32> 38 // CHECK: bufferization.alloc_tensor() : tensor<5x6xf32> 39 %2 = bufferization.alloc_tensor() : tensor<5x6xf32> 40 // CHECK: bufferization.alloc_tensor(%{{.*}}, %{{.*}}) : tensor<?x?xf32> 41 %3 = bufferization.alloc_tensor(%sz, %sz) : tensor<?x?xf32> 42 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) {escape = true} : tensor<?x5xf32> 43 %4 = bufferization.alloc_tensor() copy(%t) {escape = true} : tensor<?x5xf32> 44 // CHECK: bufferization.alloc_tensor() copy(%{{.*}}) {escape = false} : tensor<?x5xf32> 45 %5 = bufferization.alloc_tensor() copy(%t) {escape = false} : tensor<?x5xf32> 46 %c100 = arith.constant 100 : index 47 // CHECK: bufferization.alloc_tensor() size_hint= 48 %6 = bufferization.alloc_tensor() size_hint=%c100 : tensor<100x100xf64, #CSR> 49 // CHECK: bufferization.alloc_tensor(%{{.+}}) {memory_space = "foo"} : tensor<?xf32> 50 %7 = bufferization.alloc_tensor(%sz) {memory_space = "foo"} : tensor<?xf32> 51 return %1 : tensor<?x5xf32> 52} 53 54// CHECK-LABEL: func @test_dealloc_tensor_op 55func.func @test_dealloc_tensor_op(%arg0: tensor<4xi32>) { 56 // CHECK: bufferization.dealloc_tensor {{.*}} : tensor<4xi32> 57 bufferization.dealloc_tensor %arg0 : tensor<4xi32> 58 return 59} 60 61// CHECK-LABEL: func @test_materialize_in_destination_op 62func.func @test_materialize_in_destination_op( 63 %arg0: tensor<?xf32>, %arg1: tensor<?xf32>, %arg2: memref<?xf32, 3>, 64 %arg4: tensor<5xf32>) -> tensor<?xf32> { 65 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, tensor<?xf32>) -> tensor<?xf32> 66 %1 = bufferization.materialize_in_destination %arg0 in %arg1 : (tensor<?xf32>, tensor<?xf32>) -> tensor<?xf32> 67 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, memref<?xf32, 3>) -> () 68 bufferization.materialize_in_destination %arg0 in restrict writable %arg2 : (tensor<?xf32>, memref<?xf32, 3>) -> () 69 // CHECK: bufferization.materialize_in_destination {{.*}} : (tensor<?xf32>, tensor<5xf32>) -> tensor<5xf32> 70 %2 = bufferization.materialize_in_destination %arg0 in %arg4 : (tensor<?xf32>, tensor<5xf32>) -> tensor<5xf32> 71 return %1 : tensor<?xf32> 72} 73 74// CHECK-LABEL: func @test_dealloc_op 75func.func @test_dealloc_op(%arg0: memref<2xf32>, %arg1: memref<4xi32>, 76 %arg2: i1, %arg3: i1, %arg4: memref<?xf32>, 77 %arg5: memref<*xf64>) -> (i1, i1) { 78 // CHECK: bufferization.dealloc (%arg0, %arg1 : memref<2xf32>, memref<4xi32>) if (%arg2, %arg3) retain (%arg4, %arg5 : memref<?xf32>, memref<*xf64>) 79 %0:2 = bufferization.dealloc (%arg0, %arg1 : memref<2xf32>, memref<4xi32>) if (%arg2, %arg3) retain (%arg4, %arg5 : memref<?xf32>, memref<*xf64>) 80 // CHECK: bufferization.dealloc (%arg0 : memref<2xf32>) if (%arg2) 81 bufferization.dealloc (%arg0 : memref<2xf32>) if (%arg2) 82 // CHECK: bufferization.dealloc 83 bufferization.dealloc 84 return %0#0, %0#1 : i1, i1 85} 86