debug.mlir
4.96 KB
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
// RUN: mlir-translate -test-spirv-roundtrip-debug -mlir-print-debuginfo %s | FileCheck %s
spv.module Logical GLSL450 requires #spv.vce<v1.0, [Shader], []> {
// CHECK: loc({{".*debug.mlir"}}:5:3)
spv.globalVariable @var0 bind(0, 1) : !spv.ptr<f32, Input>
spv.func @arithmetic(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {
// CHECK: loc({{".*debug.mlir"}}:8:10)
%0 = spv.FAdd %arg0, %arg1 : vector<4xf32>
// CHECK: loc({{".*debug.mlir"}}:10:10)
%1 = spv.FNegate %arg0 : vector<4xf32>
spv.Return
}
spv.func @atomic(%ptr: !spv.ptr<i32, Workgroup>, %value: i32, %comparator: i32) "None" {
// CHECK: loc({{".*debug.mlir"}}:16:10)
%1 = spv.AtomicAnd "Device" "None" %ptr, %value : !spv.ptr<i32, Workgroup>
spv.Return
}
spv.func @bitwiser(%arg0 : i32, %arg1 : i32) "None" {
// CHECK: loc({{".*debug.mlir"}}:22:10)
%0 = spv.BitwiseAnd %arg0, %arg1 : i32
spv.Return
}
spv.func @convert(%arg0 : f32) "None" {
// CHECK: loc({{".*debug.mlir"}}:28:10)
%0 = spv.ConvertFToU %arg0 : f32 to i32
spv.Return
}
spv.func @composite(%arg0 : !spv.struct<f32, !spv.struct<!spv.array<4xf32>, f32>>, %arg1: !spv.array<4xf32>, %arg2 : f32, %arg3 : f32) "None" {
// CHECK: loc({{".*debug.mlir"}}:34:10)
%0 = spv.CompositeInsert %arg1, %arg0[1 : i32, 0 : i32] : !spv.array<4xf32> into !spv.struct<f32, !spv.struct<!spv.array<4xf32>, f32>>
// CHECK: loc({{".*debug.mlir"}}:36:10)
%1 = spv.CompositeConstruct %arg2, %arg3 : vector<2xf32>
spv.Return
}
spv.func @group_non_uniform(%val: f32) "None" {
// CHECK: loc({{".*debug.mlir"}}:42:10)
%0 = spv.GroupNonUniformFAdd "Workgroup" "Reduce" %val : f32
spv.Return
}
spv.func @local_var() "None" {
%zero = spv.constant 0: i32
// CHECK: loc({{".*debug.mlir"}}:49:12)
%var = spv.Variable init(%zero) : !spv.ptr<i32, Function>
spv.Return
}
spv.func @logical(%arg0: i32, %arg1: i32) "None" {
// CHECK: loc({{".*debug.mlir"}}:55:10)
%0 = spv.IEqual %arg0, %arg1 : i32
spv.Return
}
spv.func @memory_accesses(%arg0 : !spv.ptr<!spv.array<4x!spv.array<4xf32>>, StorageBuffer>, %arg1 : i32, %arg2 : i32) "None" {
// CHECK: loc({{".*debug.mlir"}}:61:10)
%2 = spv.AccessChain %arg0[%arg1, %arg2] : !spv.ptr<!spv.array<4x!spv.array<4xf32>>, StorageBuffer>, i32, i32
// CHECK: loc({{".*debug.mlir"}}:63:10)
%3 = spv.Load "StorageBuffer" %2 : f32
// CHECK: loc({{.*debug.mlir"}}:65:5)
spv.Store "StorageBuffer" %2, %3 : f32
// CHECK: loc({{".*debug.mlir"}}:67:5)
spv.Return
}
spv.func @loop(%count : i32) -> () "None" {
%zero = spv.constant 0: i32
%one = spv.constant 1: i32
%ivar = spv.Variable init(%zero) : !spv.ptr<i32, Function>
%jvar = spv.Variable init(%zero) : !spv.ptr<i32, Function>
spv.loop {
// CHECK: loc({{".*debug.mlir"}}:75:5)
spv.Branch ^header
^header:
%ival0 = spv.Load "Function" %ivar : i32
%icmp = spv.SLessThan %ival0, %count : i32
// CHECK: loc({{".*debug.mlir"}}:75:5)
spv.BranchConditional %icmp, ^body, ^merge
^body:
spv.Store "Function" %jvar, %zero : i32
spv.loop {
// CHECK: loc({{".*debug.mlir"}}:85:7)
spv.Branch ^header
^header:
%jval0 = spv.Load "Function" %jvar : i32
%jcmp = spv.SLessThan %jval0, %count : i32
// CHECK: loc({{".*debug.mlir"}}:85:7)
spv.BranchConditional %jcmp, ^body, ^merge
^body:
// CHECK: loc({{".*debug.mlir"}}:95:9)
spv.Branch ^continue
^continue:
%jval1 = spv.Load "Function" %jvar : i32
%add = spv.IAdd %jval1, %one : i32
spv.Store "Function" %jvar, %add : i32
// CHECK: loc({{".*debug.mlir"}}:101:9)
spv.Branch ^header
^merge:
// CHECK: loc({{".*debug.mlir"}}:85:7)
spv._merge
// CHECK: loc({{".*debug.mlir"}}:85:7)
}
// CHECK: loc({{".*debug.mlir"}}:108:7)
spv.Branch ^continue
^continue:
%ival1 = spv.Load "Function" %ivar : i32
%add = spv.IAdd %ival1, %one : i32
spv.Store "Function" %ivar, %add : i32
// CHECK: loc({{".*debug.mlir"}}:114:7)
spv.Branch ^header
^merge:
// CHECK: loc({{".*debug.mlir"}}:75:5)
spv._merge
// CHECK: loc({{".*debug.mlir"}}:75:5)
}
spv.Return
}
spv.func @selection(%cond: i1) -> () "None" {
%zero = spv.constant 0: i32
%one = spv.constant 1: i32
%two = spv.constant 2: i32
%var = spv.Variable init(%zero) : !spv.ptr<i32, Function>
spv.selection {
// CHECK: loc({{".*debug.mlir"}}:128:5)
spv.BranchConditional %cond [5, 10], ^then, ^else
^then:
spv.Store "Function" %var, %one : i32
// CHECK: loc({{".*debug.mlir"}}:134:7)
spv.Branch ^merge
^else:
spv.Store "Function" %var, %two : i32
// CHECK: loc({{".*debug.mlir"}}:138:7)
spv.Branch ^merge
^merge:
// CHECK: loc({{".*debug.mlir"}}:128:5)
spv._merge
// CHECK: loc({{".*debug.mlir"}}:128:5)
}
spv.Return
}
}