1// RUN: mlir-translate -split-input-file -test-spirv-roundtrip %s | FileCheck %s 2 3// Single loop 4 5spv.module Logical GLSL450 requires #spv.vce<v1.0, [Shader], []> { 6 // for (int i = 0; i < count; ++i) {} 7// CHECK-LABEL: @loop 8 spv.func @loop(%count : i32) -> () "None" { 9 %zero = spv.Constant 0: i32 10 %one = spv.Constant 1: i32 11 %var = spv.Variable init(%zero) : !spv.ptr<i32, Function> 12 13// CHECK: spv.Branch ^bb1 14// CHECK-NEXT: ^bb1: 15// CHECK-NEXT: spv.mlir.loop 16 spv.mlir.loop { 17// CHECK-NEXT: spv.Branch ^bb1 18 spv.Branch ^header 19 20// CHECK-NEXT: ^bb1: 21 ^header: 22// CHECK-NEXT: spv.Load 23 %val0 = spv.Load "Function" %var : i32 24// CHECK-NEXT: spv.SLessThan 25 %cmp = spv.SLessThan %val0, %count : i32 26// CHECK-NEXT: spv.BranchConditional %{{.*}} [1, 1], ^bb2, ^bb4 27 spv.BranchConditional %cmp [1, 1], ^body, ^merge 28 29// CHECK-NEXT: ^bb2: 30 ^body: 31 // Do nothing 32// CHECK-NEXT: spv.Branch ^bb3 33 spv.Branch ^continue 34 35// CHECK-NEXT: ^bb3: 36 ^continue: 37// CHECK-NEXT: spv.Load 38 %val1 = spv.Load "Function" %var : i32 39// CHECK-NEXT: spv.Constant 1 40// CHECK-NEXT: spv.IAdd 41 %add = spv.IAdd %val1, %one : i32 42// CHECK-NEXT: spv.Store 43 spv.Store "Function" %var, %add : i32 44// CHECK-NEXT: spv.Branch ^bb1 45 spv.Branch ^header 46 47// CHECK-NEXT: ^bb4: 48// CHECK-NEXT: spv.mlir.merge 49 ^merge: 50 spv.mlir.merge 51 } 52 spv.Return 53 } 54 55 spv.func @main() -> () "None" { 56 spv.Return 57 } 58 spv.EntryPoint "GLCompute" @main 59} 60 61// ----- 62 63// Single loop with block arguments 64 65spv.module Logical GLSL450 requires #spv.vce<v1.0, [Shader], []> { 66 spv.GlobalVariable @GV1 bind(0, 0) : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer> 67 spv.GlobalVariable @GV2 bind(0, 1) : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer> 68// CHECK-LABEL: @loop_kernel 69 spv.func @loop_kernel() "None" { 70 %0 = spv.mlir.addressof @GV1 : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer> 71 %1 = spv.Constant 0 : i32 72 %2 = spv.AccessChain %0[%1] : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer>, i32 73 %3 = spv.mlir.addressof @GV2 : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer> 74 %5 = spv.AccessChain %3[%1] : !spv.ptr<!spv.struct<(!spv.array<10 x f32, stride=4> [0])>, StorageBuffer>, i32 75 %6 = spv.Constant 4 : i32 76 %7 = spv.Constant 42 : i32 77 %8 = spv.Constant 2 : i32 78// CHECK: spv.Branch ^bb1(%{{.*}} : i32) 79// CHECK-NEXT: ^bb1(%[[OUTARG:.*]]: i32): 80// CHECK-NEXT: spv.mlir.loop { 81 spv.mlir.loop { 82// CHECK-NEXT: spv.Branch ^bb1(%[[OUTARG]] : i32) 83 spv.Branch ^header(%6 : i32) 84// CHECK-NEXT: ^bb1(%[[HEADARG:.*]]: i32): 85 ^header(%9: i32): 86 %10 = spv.SLessThan %9, %7 : i32 87// CHECK: spv.BranchConditional %{{.*}}, ^bb2, ^bb3 88 spv.BranchConditional %10, ^body, ^merge 89// CHECK-NEXT: ^bb2: // pred: ^bb1 90 ^body: 91 %11 = spv.AccessChain %2[%9] : !spv.ptr<!spv.array<10 x f32, stride=4>, StorageBuffer>, i32 92 %12 = spv.Load "StorageBuffer" %11 : f32 93 %13 = spv.AccessChain %5[%9] : !spv.ptr<!spv.array<10 x f32, stride=4>, StorageBuffer>, i32 94 spv.Store "StorageBuffer" %13, %12 : f32 95// CHECK: %[[ADD:.*]] = spv.IAdd 96 %14 = spv.IAdd %9, %8 : i32 97// CHECK-NEXT: spv.Branch ^bb1(%[[ADD]] : i32) 98 spv.Branch ^header(%14 : i32) 99// CHECK-NEXT: ^bb3: 100 ^merge: 101// CHECK-NEXT: spv.mlir.merge 102 spv.mlir.merge 103 } 104 spv.Return 105 } 106 spv.EntryPoint "GLCompute" @loop_kernel 107 spv.ExecutionMode @loop_kernel "LocalSize", 1, 1, 1 108} 109 110// ----- 111 112// Nested loop 113 114spv.module Logical GLSL450 requires #spv.vce<v1.0, [Shader], []> { 115 // for (int i = 0; i < count; ++i) { 116 // for (int j = 0; j < count; ++j) { } 117 // } 118// CHECK-LABEL: @loop 119 spv.func @loop(%count : i32) -> () "None" { 120 %zero = spv.Constant 0: i32 121 %one = spv.Constant 1: i32 122 %ivar = spv.Variable init(%zero) : !spv.ptr<i32, Function> 123 %jvar = spv.Variable init(%zero) : !spv.ptr<i32, Function> 124 125// CHECK: spv.Branch ^bb1 126// CHECK-NEXT: ^bb1: 127// CHECK-NEXT: spv.mlir.loop control(Unroll) 128 spv.mlir.loop control(Unroll) { 129// CHECK-NEXT: spv.Branch ^bb1 130 spv.Branch ^header 131 132// CHECK-NEXT: ^bb1: 133 ^header: 134// CHECK-NEXT: spv.Load 135 %ival0 = spv.Load "Function" %ivar : i32 136// CHECK-NEXT: spv.SLessThan 137 %icmp = spv.SLessThan %ival0, %count : i32 138// CHECK-NEXT: spv.BranchConditional %{{.*}}, ^bb2, ^bb5 139 spv.BranchConditional %icmp, ^body, ^merge 140 141// CHECK-NEXT: ^bb2: 142 ^body: 143// CHECK-NEXT: spv.Constant 0 144// CHECK-NEXT: spv.Store 145 spv.Store "Function" %jvar, %zero : i32 146// CHECK-NEXT: spv.Branch ^bb3 147// CHECK-NEXT: ^bb3: 148// CHECK-NEXT: spv.mlir.loop control(DontUnroll) 149 spv.mlir.loop control(DontUnroll) { 150// CHECK-NEXT: spv.Branch ^bb1 151 spv.Branch ^header 152 153// CHECK-NEXT: ^bb1: 154 ^header: 155// CHECK-NEXT: spv.Load 156 %jval0 = spv.Load "Function" %jvar : i32 157// CHECK-NEXT: spv.SLessThan 158 %jcmp = spv.SLessThan %jval0, %count : i32 159// CHECK-NEXT: spv.BranchConditional %{{.*}}, ^bb2, ^bb4 160 spv.BranchConditional %jcmp, ^body, ^merge 161 162// CHECK-NEXT: ^bb2: 163 ^body: 164 // Do nothing 165// CHECK-NEXT: spv.Branch ^bb3 166 spv.Branch ^continue 167 168// CHECK-NEXT: ^bb3: 169 ^continue: 170// CHECK-NEXT: spv.Load 171 %jval1 = spv.Load "Function" %jvar : i32 172// CHECK-NEXT: spv.Constant 1 173// CHECK-NEXT: spv.IAdd 174 %add = spv.IAdd %jval1, %one : i32 175// CHECK-NEXT: spv.Store 176 spv.Store "Function" %jvar, %add : i32 177// CHECK-NEXT: spv.Branch ^bb1 178 spv.Branch ^header 179 180// CHECK-NEXT: ^bb4: 181 ^merge: 182// CHECK-NEXT: spv.mlir.merge 183 spv.mlir.merge 184 } // end inner loop 185 186// CHECK: spv.Branch ^bb4 187 spv.Branch ^continue 188 189// CHECK-NEXT: ^bb4: 190 ^continue: 191// CHECK-NEXT: spv.Load 192 %ival1 = spv.Load "Function" %ivar : i32 193// CHECK-NEXT: spv.Constant 1 194// CHECK-NEXT: spv.IAdd 195 %add = spv.IAdd %ival1, %one : i32 196// CHECK-NEXT: spv.Store 197 spv.Store "Function" %ivar, %add : i32 198// CHECK-NEXT: spv.Branch ^bb1 199 spv.Branch ^header 200 201// CHECK-NEXT: ^bb5: 202// CHECK-NEXT: spv.mlir.merge 203 ^merge: 204 spv.mlir.merge 205 } // end outer loop 206 spv.Return 207 } 208 209 spv.func @main() -> () "None" { 210 spv.Return 211 } 212 spv.EntryPoint "GLCompute" @main 213} 214 215 216// ----- 217 218// Loop with selection in its header 219 220spv.module Physical64 OpenCL requires #spv.vce<v1.0, [Kernel, Linkage, Addresses, Int64], []> { 221// CHECK-LABEL: @kernel 222// CHECK-SAME: (%[[INPUT0:.+]]: i64) 223 spv.func @kernel(%input: i64) "None" { 224// CHECK-NEXT: %[[VAR:.+]] = spv.Variable : !spv.ptr<i1, Function> 225// CHECK-NEXT: spv.Branch ^[[BB0:.+]](%[[INPUT0]] : i64) 226// CHECK-NEXT: ^[[BB0]](%[[INPUT1:.+]]: i64): 227 %cst0_i64 = spv.Constant 0 : i64 228 %true = spv.Constant true 229 %false = spv.Constant false 230// CHECK-NEXT: spv.mlir.loop { 231 spv.mlir.loop { 232// CHECK-NEXT: spv.Branch ^[[LOOP_HEADER:.+]](%[[INPUT1]] : i64) 233 spv.Branch ^loop_header(%input : i64) 234// CHECK-NEXT: ^[[LOOP_HEADER]](%[[ARG1:.+]]: i64): 235 ^loop_header(%arg1: i64): 236// CHECK-NEXT: spv.Branch ^[[LOOP_BODY:.+]] 237// CHECK-NEXT: ^[[LOOP_BODY]]: 238// CHECK-NEXT: %[[C0:.+]] = spv.Constant 0 : i64 239 %gt = spv.SGreaterThan %arg1, %cst0_i64 : i64 240// CHECK-NEXT: %[[GT:.+]] = spv.SGreaterThan %[[ARG1]], %[[C0]] : i64 241// CHECK-NEXT: spv.Branch ^[[BB1:.+]] 242// CHECK-NEXT: ^[[BB1]]: 243 %var = spv.Variable : !spv.ptr<i1, Function> 244// CHECK-NEXT: spv.mlir.selection { 245 spv.mlir.selection { 246// CHECK-NEXT: spv.BranchConditional %[[GT]], ^[[THEN:.+]], ^[[ELSE:.+]] 247 spv.BranchConditional %gt, ^then, ^else 248// CHECK-NEXT: ^[[THEN]]: 249 ^then: 250// CHECK-NEXT: %true = spv.Constant true 251// CHECK-NEXT: spv.Store "Function" %[[VAR]], %true : i1 252 spv.Store "Function" %var, %true : i1 253// CHECK-NEXT: spv.Branch ^[[SELECTION_MERGE:.+]] 254 spv.Branch ^selection_merge 255// CHECK-NEXT: ^[[ELSE]]: 256 ^else: 257// CHECK-NEXT: %false = spv.Constant false 258// CHECK-NEXT: spv.Store "Function" %[[VAR]], %false : i1 259 spv.Store "Function" %var, %false : i1 260// CHECK-NEXT: spv.Branch ^[[SELECTION_MERGE]] 261 spv.Branch ^selection_merge 262// CHECK-NEXT: ^[[SELECTION_MERGE]]: 263 ^selection_merge: 264// CHECK-NEXT: spv.mlir.merge 265 spv.mlir.merge 266// CHECK-NEXT: } 267 } 268// CHECK-NEXT: %[[LOAD:.+]] = spv.Load "Function" %[[VAR]] : i1 269 %load = spv.Load "Function" %var : i1 270// CHECK-NEXT: spv.BranchConditional %[[LOAD]], ^[[CONTINUE:.+]](%[[ARG1]] : i64), ^[[LOOP_MERGE:.+]] 271 spv.BranchConditional %load, ^continue(%arg1 : i64), ^loop_merge 272// CHECK-NEXT: ^[[CONTINUE]](%[[ARG2:.+]]: i64): 273 ^continue(%arg2: i64): 274// CHECK-NEXT: %[[C0:.+]] = spv.Constant 0 : i64 275// CHECK-NEXT: %[[LT:.+]] = spv.SLessThan %[[ARG2]], %[[C0]] : i64 276 %lt = spv.SLessThan %arg2, %cst0_i64 : i64 277// CHECK-NEXT: spv.Store "Function" %[[VAR]], %[[LT]] : i1 278 spv.Store "Function" %var, %lt : i1 279// CHECK-NEXT: spv.Branch ^[[LOOP_HEADER]](%[[ARG2]] : i64) 280 spv.Branch ^loop_header(%arg2 : i64) 281// CHECK-NEXT: ^[[LOOP_MERGE]]: 282 ^loop_merge: 283// CHECK-NEXT: spv.mlir.merge 284 spv.mlir.merge 285// CHECK-NEXT: } 286 } 287// CHECK-NEXT: spv.Return 288 spv.Return 289 } 290} 291