1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// RUN: mlir-opt --transform-interpreter=entry-point=all %s | FileCheck %s --check-prefixes=CHECK,ALL
// RUN: mlir-opt --transform-interpreter=entry-point=small %s | FileCheck %s --check-prefixes=CHECK,SMALL
func.func private @callee(memref<*xf32>)
// CHECK-LABEL: @large_alloc
func.func @large_alloc() {
// SMALL: memref.alloc()
// ALL: memref.alloca
%0 = memref.alloc() : memref<100x100xf32>
%1 = memref.cast %0 : memref<100x100xf32> to memref<*xf32>
call @callee(%1) : (memref<*xf32>) -> ()
// SMALL: memref.dealloc
// ALL-NOT: memref.dealloc
memref.dealloc %0 : memref<100x100xf32>
return
}
// CHECK-LABEL: @small_alloc
func.func @small_alloc() {
// CHECK: memref.alloca
%0 = memref.alloc() : memref<2x2xf32>
%1 = memref.cast %0 : memref<2x2xf32> to memref<*xf32>
call @callee(%1) : (memref<*xf32>) -> ()
// CHECK-NOT: memref.dealloc
memref.dealloc %0 : memref<2x2xf32>
return
}
// CHECK-LABEL: @no_dealloc
func.func @no_dealloc() {
// CHECK: memref.alloc()
%0 = memref.alloc() : memref<2x2xf32>
%1 = memref.cast %0 : memref<2x2xf32> to memref<*xf32>
call @callee(%1) : (memref<*xf32>) -> ()
return
}
// CHECK-LABEL: @mismatching_scope
func.func @mismatching_scope() {
// CHECK: memref.alloc()
%0 = memref.alloc() : memref<2x2xf32>
%1 = memref.cast %0 : memref<2x2xf32> to memref<*xf32>
call @callee(%1) : (memref<*xf32>) -> ()
scf.execute_region {
memref.dealloc %0 : memref<2x2xf32>
scf.yield
}
return
}
module attributes {transform.with_named_sequence} {
transform.named_sequence @all(%arg0: !transform.any_op {transform.readonly}) {
%0 = transform.structured.match ops{["func.func"]} in %arg0 : (!transform.any_op) -> !transform.any_op
transform.apply_patterns to %0 {
transform.apply_patterns.memref.alloc_to_alloca
} : !transform.any_op
transform.yield
}
}
module attributes {transform.with_named_sequence} {
transform.named_sequence @small(%arg0: !transform.any_op {transform.readonly}) {
%0 = transform.structured.match ops{["func.func"]} in %arg0 : (!transform.any_op) -> !transform.any_op
transform.apply_patterns to %0 {
transform.apply_patterns.memref.alloc_to_alloca size_limit(32)
} : !transform.any_op
transform.yield
}
}
|
