flang/test/Lower/OpenACC/acc-kernels-loop.f90

   1 ! This test checks lowering of OpenACC kernels loop combined directive.
   2
   3 ! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s
   4
   5 subroutine acc_kernels_loop
   6   integer :: i, j
   7
   8   integer :: async = 1
   9   integer :: wait1 = 1
  10   integer :: wait2 = 2
  11   integer :: numGangs = 1
  12   integer :: numWorkers = 10
  13   integer :: vectorLength = 128
  14   logical :: ifCondition = .TRUE.
  15   integer, parameter :: n = 10
  16   real, dimension(n) :: a, b, c
  17   real, dimension(n, n) :: d, e
  18   real, pointer :: f, g
  19   integer :: reduction_i
  20   real :: reduction_r
  21
  22   integer :: gangNum = 8
  23   integer :: gangStatic = 8
  24   integer :: vectorNum = 128
  25   integer, parameter :: tileSize = 2
  26
  27 ! CHECK: %[[A:.*]] = fir.alloca !fir.array<10xf32> {{{.*}}uniq_name = "{{.*}}Ea"}
  28 ! CHECK: %[[DECLA:.*]]:2 = hlfir.declare %[[A]]
  29 ! CHECK: %[[B:.*]] = fir.alloca !fir.array<10xf32> {{{.*}}uniq_name = "{{.*}}Eb"}
  30 ! CHECK: %[[DECLB:.*]]:2 = hlfir.declare %[[B]]
  31 ! CHECK: %[[C:.*]] = fir.alloca !fir.array<10xf32> {{{.*}}uniq_name = "{{.*}}Ec"}
  32 ! CHECK: %[[DECLC:.*]]:2 = hlfir.declare %[[C]]
  33 ! CHECK: %[[F:.*]] = fir.alloca !fir.box<!fir.ptr<f32>> {bindc_name = "f", uniq_name = "{{.*}}Ef"}
  34 ! CHECK: %[[DECLF:.*]]:2 = hlfir.declare %[[F]]
  35 ! CHECK: %[[G:.*]] = fir.alloca !fir.box<!fir.ptr<f32>> {bindc_name = "g", uniq_name = "{{.*}}Eg"}
  36 ! CHECK: %[[DECLG:.*]]:2 = hlfir.declare %[[G]]
  37 ! CHECK: %[[IFCONDITION:.*]] = fir.address_of(@{{.*}}ifcondition) : !fir.ref<!fir.logical<4>>
  38 ! CHECK: %[[DECLIFCONDITION:.*]]:2 = hlfir.declare %[[IFCONDITION]]
  39
  40   !$acc kernels
  41   !$acc loop
  42   DO i = 1, n
  43     a(i) = b(i)
  44   END DO
  45   !$acc end kernels
  46
  47 ! CHECK:      acc.kernels {
  48 ! CHECK:        acc.loop private{{.*}} {
  49 ! CHECK:          acc.yield
  50 ! CHECK-NEXT:   }{{$}}
  51 ! CHECK:        acc.terminator
  52 ! CHECK-NEXT: }{{$}}
  53
  54   !$acc kernels loop
  55   DO i = 1, n
  56     a(i) = b(i)
  57   END DO
  58
  59 ! CHECK:      acc.kernels combined(loop) {
  60 ! CHECK:        acc.loop combined(kernels) private{{.*}} {
  61 ! CHECK:          acc.yield
  62 ! CHECK-NEXT:   }{{$}}
  63 ! CHECK:        acc.terminator
  64 ! CHECK-NEXT: }{{$}}
  65
  66   !$acc kernels loop async
  67   DO i = 1, n
  68     a(i) = b(i)
  69   END DO
  70   !$acc end kernels loop
  71
  72 ! CHECK:      acc.kernels {{.*}} {
  73 ! CHECK:        acc.loop {{.*}} {
  74 ! CHECK:          acc.yield
  75 ! CHECK-NEXT:   }{{$}}
  76 ! CHECK:        acc.terminator
  77 ! CHECK-NEXT: } attributes {asyncOnly = [#acc.device_type<none>]}
  78
  79   !$acc kernels loop async(1)
  80   DO i = 1, n
  81     a(i) = b(i)
  82   END DO
  83
  84 ! CHECK:      [[ASYNC1:%.*]] = arith.constant 1 : i32
  85 ! CHECK:      acc.kernels {{.*}} async([[ASYNC1]] : i32) {
  86 ! CHECK:        acc.loop {{.*}} {
  87 ! CHECK:          acc.yield
  88 ! CHECK-NEXT:   }{{$}}
  89 ! CHECK:        acc.terminator
  90 ! CHECK-NEXT: }{{$}}
  91
  92   !$acc kernels loop async(async)
  93   DO i = 1, n
  94     a(i) = b(i)
  95   END DO
  96
  97 ! CHECK:      [[ASYNC2:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
  98 ! CHECK:      acc.kernels {{.*}} async([[ASYNC2]] : i32) {
  99 ! CHECK:        acc.loop {{.*}} {
 100 ! CHECK:          acc.yield
 101 ! CHECK-NEXT:   }{{$}}
 102 ! CHECK:        acc.terminator
 103 ! CHECK-NEXT: }{{$}}
 104
 105   !$acc kernels loop wait
 106   DO i = 1, n
 107     a(i) = b(i)
 108   END DO
 109
 110 ! CHECK:      acc.kernels {{.*}} wait {
 111 ! CHECK:        acc.loop {{.*}} {
 112 ! CHECK:          acc.yield
 113 ! CHECK-NEXT:   }{{$}}
 114 ! CHECK:        acc.terminator
 115 ! CHECK-NEXT: }
 116
 117   !$acc kernels loop wait(1)
 118   DO i = 1, n
 119     a(i) = b(i)
 120   END DO
 121
 122 ! CHECK:      [[WAIT1:%.*]] = arith.constant 1 : i32
 123 ! CHECK:      acc.kernels {{.*}} wait({[[WAIT1]] : i32}) {
 124 ! CHECK:        acc.loop
 125 ! CHECK:          acc.yield
 126 ! CHECK-NEXT:   }{{$}}
 127 ! CHECK:        acc.terminator
 128 ! CHECK-NEXT: }{{$}}
 129
 130   !$acc kernels loop wait(1, 2)
 131   DO i = 1, n
 132     a(i) = b(i)
 133   END DO
 134
 135 ! CHECK:      [[WAIT2:%.*]] = arith.constant 1 : i32
 136 ! CHECK:      [[WAIT3:%.*]] = arith.constant 2 : i32
 137 ! CHECK:      acc.kernels {{.*}} wait({[[WAIT2]] : i32, [[WAIT3]] : i32}) {
 138 ! CHECK:        acc.loop
 139 ! CHECK:          acc.yield
 140 ! CHECK-NEXT:   }{{$}}
 141 ! CHECK:        acc.terminator
 142 ! CHECK-NEXT: }{{$}}
 143
 144   !$acc kernels loop wait(wait1, wait2)
 145   DO i = 1, n
 146     a(i) = b(i)
 147   END DO
 148
 149 ! CHECK:      [[WAIT4:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 150 ! CHECK:      [[WAIT5:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 151 ! CHECK:      acc.kernels {{.*}} wait({[[WAIT4]] : i32, [[WAIT5]] : i32}) {
 152 ! CHECK:        acc.loop
 153 ! CHECK:          acc.yield
 154 ! CHECK-NEXT:   }{{$}}
 155 ! CHECK:        acc.terminator
 156 ! CHECK-NEXT: }{{$}}
 157
 158   !$acc kernels loop num_gangs(1)
 159   DO i = 1, n
 160     a(i) = b(i)
 161   END DO
 162
 163 ! CHECK:      [[NUMGANGS1:%.*]] = arith.constant 1 : i32
 164 ! CHECK:      acc.kernels {{.*}} num_gangs({[[NUMGANGS1]] : i32}) {
 165 ! CHECK:        acc.loop
 166 ! CHECK:          acc.yield
 167 ! CHECK-NEXT:   }{{$}}
 168 ! CHECK:        acc.terminator
 169 ! CHECK-NEXT: }{{$}}
 170
 171   !$acc kernels loop num_gangs(numGangs)
 172   DO i = 1, n
 173     a(i) = b(i)
 174   END DO
 175
 176 ! CHECK:      [[NUMGANGS2:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 177 ! CHECK:      acc.kernels {{.*}} num_gangs({[[NUMGANGS2]] : i32}) {
 178 ! CHECK:        acc.loop
 179 ! CHECK:          acc.yield
 180 ! CHECK-NEXT:   }{{$}}
 181 ! CHECK:        acc.terminator
 182 ! CHECK-NEXT: }{{$}}
 183
 184   !$acc kernels loop num_workers(10)
 185   DO i = 1, n
 186     a(i) = b(i)
 187   END DO
 188
 189 ! CHECK:      [[NUMWORKERS1:%.*]] = arith.constant 10 : i32
 190 ! CHECK:      acc.kernels {{.*}} num_workers([[NUMWORKERS1]] : i32) {
 191 ! CHECK:        acc.loop {{.*}} {
 192 ! CHECK:          acc.yield
 193 ! CHECK-NEXT:   }{{$}}
 194 ! CHECK:        acc.terminator
 195 ! CHECK-NEXT: }{{$}}
 196
 197   !$acc kernels loop num_workers(numWorkers)
 198   DO i = 1, n
 199     a(i) = b(i)
 200   END DO
 201
 202 ! CHECK:      [[NUMWORKERS2:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 203 ! CHECK:      acc.kernels {{.*}} num_workers([[NUMWORKERS2]] : i32) {
 204 ! CHECK:        acc.loop {{.*}} {
 205 ! CHECK:          acc.yield
 206 ! CHECK-NEXT:   }{{$}}
 207 ! CHECK:        acc.terminator
 208 ! CHECK-NEXT: }{{$}}
 209
 210   !$acc kernels loop vector_length(128)
 211   DO i = 1, n
 212     a(i) = b(i)
 213   END DO
 214
 215 ! CHECK:      [[VECTORLENGTH1:%.*]] = arith.constant 128 : i32
 216 ! CHECK:      acc.kernels {{.*}} vector_length([[VECTORLENGTH1]] : i32) {
 217 ! CHECK:        acc.loop {{.*}} {
 218 ! CHECK:          acc.yield
 219 ! CHECK-NEXT:   }{{$}}
 220 ! CHECK:        acc.terminator
 221 ! CHECK-NEXT: }{{$}}
 222
 223   !$acc kernels loop vector_length(vectorLength)
 224   DO i = 1, n
 225     a(i) = b(i)
 226   END DO
 227
 228 ! CHECK:      [[VECTORLENGTH2:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 229 ! CHECK:      acc.kernels {{.*}} vector_length([[VECTORLENGTH2]] : i32) {
 230 ! CHECK:        acc.loop {{.*}} {
 231 ! CHECK:          acc.yield
 232 ! CHECK-NEXT:   }{{$}}
 233 ! CHECK:        acc.terminator
 234 ! CHECK-NEXT: }{{$}}
 235
 236   !$acc kernels loop if(.TRUE.)
 237   DO i = 1, n
 238     a(i) = b(i)
 239   END DO
 240
 241 ! CHECK:      [[IF1:%.*]] = arith.constant true
 242 ! CHECK:      acc.kernels {{.*}} if([[IF1]]) {
 243 ! CHECK:        acc.loop {{.*}} {
 244 ! CHECK:          acc.yield
 245 ! CHECK-NEXT:   }{{$}}
 246 ! CHECK:        acc.terminator
 247 ! CHECK-NEXT: }{{$}}
 248
 249   !$acc kernels loop if(ifCondition)
 250   DO i = 1, n
 251     a(i) = b(i)
 252   END DO
 253
 254 ! CHECK:      [[IFCOND:%.*]] = fir.load %{{.*}} : !fir.ref<!fir.logical<4>>
 255 ! CHECK:      [[IF2:%.*]] = fir.convert [[IFCOND]] : (!fir.logical<4>) -> i1
 256 ! CHECK:      acc.kernels {{.*}} if([[IF2]]) {
 257 ! CHECK:        acc.loop {{.*}} {
 258 ! CHECK:          acc.yield
 259 ! CHECK-NEXT:   }{{$}}
 260 ! CHECK:        acc.terminator
 261 ! CHECK-NEXT: }{{$}}
 262
 263   !$acc kernels loop self(.TRUE.)
 264   DO i = 1, n
 265     a(i) = b(i)
 266   END DO
 267
 268 ! CHECK:      [[SELF1:%.*]] = arith.constant true
 269 ! CHECK:      acc.kernels {{.*}} self([[SELF1]]) {
 270 ! CHECK:        acc.loop {{.*}} {
 271 ! CHECK:          acc.yield
 272 ! CHECK-NEXT:   }{{$}}
 273 ! CHECK:        acc.terminator
 274 ! CHECK-NEXT: }{{$}}
 275
 276   !$acc kernels loop self
 277   DO i = 1, n
 278     a(i) = b(i)
 279   END DO
 280
 281 ! CHECK:      acc.kernels {{.*}}{
 282 ! CHECK:        acc.loop {{.*}} {
 283 ! CHECK:          acc.yield
 284 ! CHECK-NEXT:   }{{$}}
 285 ! CHECK:        acc.terminator
 286 ! CHECK-NEXT: } attributes {selfAttr}
 287
 288   !$acc kernels loop self(ifCondition)
 289   DO i = 1, n
 290     a(i) = b(i)
 291   END DO
 292
 293
 294 ! CHECK:      %[[SELF2:.*]] = fir.convert %[[DECLIFCONDITION]]#1 : (!fir.ref<!fir.logical<4>>) -> i1
 295 ! CHECK:      acc.kernels {{.*}} self(%[[SELF2]]) {
 296 ! CHECK:        acc.loop {{.*}} {
 297 ! CHECK:          acc.yield
 298 ! CHECK-NEXT:   }{{$}}
 299 ! CHECK:        acc.terminator
 300 ! CHECK-NEXT: }{{$}}
 301
 302   !$acc kernels loop copy(a, b)
 303   DO i = 1, n
 304     a(i) = b(i)
 305   END DO
 306
 307
 308 ! CHECK:      %[[COPYIN_A:.*]] = acc.copyin varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copy>, name = "a"}
 309 ! CHECK:      %[[COPYIN_B:.*]] = acc.copyin varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copy>, name = "b"}
 310 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[COPYIN_A]], %[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 311 ! CHECK:        acc.loop {{.*}} {
 312 ! CHECK:          acc.yield
 313 ! CHECK-NEXT:   }{{$}}
 314 ! CHECK:        acc.terminator
 315 ! CHECK-NEXT: }{{$}}
 316 ! CHECK:      acc.copyout accPtr(%[[COPYIN_A]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copy>, name = "a"}
 317 ! CHECK:      acc.copyout accPtr(%[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copy>, name = "b"}
 318
 319   !$acc kernels loop copy(a) copy(b)
 320   DO i = 1, n
 321     a(i) = b(i)
 322   END DO
 323
 324 ! CHECK:      %[[COPYIN_A:.*]] = acc.copyin varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copy>, name = "a"}
 325 ! CHECK:      %[[COPYIN_B:.*]] = acc.copyin varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copy>, name = "b"}
 326 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[COPYIN_A]], %[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 327 ! CHECK:        acc.loop {{.*}} {
 328 ! CHECK:          acc.yield
 329 ! CHECK-NEXT:   }{{$}}
 330 ! CHECK:        acc.terminator
 331 ! CHECK-NEXT: }{{$}}
 332 ! CHECK:      acc.copyout accPtr(%[[COPYIN_A]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copy>, name = "a"}
 333 ! CHECK:      acc.copyout accPtr(%[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copy>, name = "b"}
 334
 335   !$acc kernels loop copyin(a) copyin(readonly: b)
 336   DO i = 1, n
 337     a(i) = b(i)
 338   END DO
 339
 340 ! CHECK:      %[[COPYIN_A:.*]] = acc.copyin varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "a"}
 341 ! CHECK:      %[[COPYIN_B:.*]] = acc.copyin varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copyin_readonly>, name = "b"}
 342 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[COPYIN_A]], %[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 343 ! CHECK:        acc.loop {{.*}} {
 344 ! CHECK:          acc.yield
 345 ! CHECK-NEXT:   }{{$}}
 346 ! CHECK:        acc.terminator
 347 ! CHECK-NEXT: }{{$}}
 348 ! CHECK:      acc.delete accPtr(%[[COPYIN_A]] : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copyin>, name = "a"}
 349 ! CHECK:      acc.delete accPtr(%[[COPYIN_B]] : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_copyin_readonly>, name = "b"}
 350
 351   !$acc kernels loop copyout(a) copyout(zero: b)
 352   DO i = 1, n
 353     a(i) = b(i)
 354   END DO
 355
 356 ! CHECK:      %[[CREATE_A:.*]] = acc.create varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copyout>, name = "a"}
 357 ! CHECK:      %[[CREATE_B:.*]] = acc.create varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_copyout>, name = "b"}
 358 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[CREATE_A]], %[[CREATE_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 359 ! CHECK:        acc.loop {{.*}} {
 360 ! CHECK:          acc.yield
 361 ! CHECK-NEXT:   }{{$}}
 362 ! CHECK:        acc.terminator
 363 ! CHECK-NEXT: }{{$}}
 364 ! CHECK:      acc.copyout accPtr(%[[CREATE_A]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) {name = "a"}
 365 ! CHECK:      acc.copyout accPtr(%[[CREATE_B]] : !fir.ref<!fir.array<10xf32>>) to varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) {name = "b"}
 366
 367   !$acc kernels loop create(b) create(zero: a)
 368   DO i = 1, n
 369     a(i) = b(i)
 370   END DO
 371
 372 ! CHECK:      %[[CREATE_B:.*]] = acc.create varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "b"}
 373 ! CHECK:      %[[CREATE_A:.*]] = acc.create varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {dataClause = #acc<data_clause acc_create_zero>, name = "a"}
 374 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[CREATE_B]], %[[CREATE_A]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 375 ! CHECK:        acc.loop {{.*}} {
 376 ! CHECK:          acc.yield
 377 ! CHECK-NEXT:   }{{$}}
 378 ! CHECK:        acc.terminator
 379 ! CHECK-NEXT: }{{$}}
 380 ! CHECK:      acc.delete accPtr(%[[CREATE_B]] : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_create>, name = "b"}
 381 ! CHECK:      acc.delete accPtr(%[[CREATE_A]] : !fir.ref<!fir.array<10xf32>>) {dataClause = #acc<data_clause acc_create_zero>, name = "a"}
 382
 383   !$acc kernels loop no_create(a, b)
 384   DO i = 1, n
 385     a(i) = b(i)
 386   END DO
 387
 388 ! CHECK:      %[[NOCREATE_A:.*]] = acc.nocreate varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "a"}
 389 ! CHECK:      %[[NOCREATE_B:.*]] = acc.nocreate varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "b"}
 390 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[NOCREATE_A]], %[[NOCREATE_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 391 ! CHECK:        acc.loop {{.*}} {
 392 ! CHECK:          acc.yield
 393 ! CHECK-NEXT:   }{{$}}
 394 ! CHECK:        acc.terminator
 395 ! CHECK-NEXT: }{{$}}
 396
 397   !$acc kernels loop present(a, b)
 398   DO i = 1, n
 399     a(i) = b(i)
 400   END DO
 401
 402 ! CHECK:      %[[PRESENT_A:.*]] = acc.present varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "a"}
 403 ! CHECK:      %[[PRESENT_B:.*]] = acc.present varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "b"}
 404 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[PRESENT_A]], %[[PRESENT_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 405 ! CHECK:        acc.loop {{.*}} {
 406 ! CHECK:          acc.yield
 407 ! CHECK-NEXT:   }{{$}}
 408 ! CHECK:        acc.terminator
 409 ! CHECK-NEXT: }{{$}}
 410
 411   !$acc kernels loop deviceptr(a) deviceptr(b)
 412   DO i = 1, n
 413     a(i) = b(i)
 414   END DO
 415
 416 ! CHECK:      %[[DEVICEPTR_A:.*]] = acc.deviceptr varPtr(%[[DECLA]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "a"}
 417 ! CHECK:      %[[DEVICEPTR_B:.*]] = acc.deviceptr varPtr(%[[DECLB]]#0 : !fir.ref<!fir.array<10xf32>>) -> !fir.ref<!fir.array<10xf32>> {name = "b"}
 418 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[DEVICEPTR_A]], %[[DEVICEPTR_B]] : !fir.ref<!fir.array<10xf32>>, !fir.ref<!fir.array<10xf32>>) {
 419 ! CHECK:        acc.loop {{.*}} {
 420 ! CHECK:          acc.yield
 421 ! CHECK-NEXT:   }{{$}}
 422 ! CHECK:        acc.terminator
 423 ! CHECK-NEXT: }{{$}}
 424
 425   !$acc kernels loop attach(f, g)
 426   DO i = 1, n
 427     a(i) = b(i)
 428   END DO
 429
 430 ! CHECK:      %[[ATTACH_F:.*]] = acc.attach varPtr(%[[DECLF]]#0 : !fir.ref<!fir.box<!fir.ptr<f32>>>) -> !fir.ref<!fir.box<!fir.ptr<f32>>> {name = "f"}
 431 ! CHECK:      %[[ATTACH_G:.*]] = acc.attach varPtr(%[[DECLG]]#0 : !fir.ref<!fir.box<!fir.ptr<f32>>>) -> !fir.ref<!fir.box<!fir.ptr<f32>>> {name = "g"}
 432 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[ATTACH_F]], %[[ATTACH_G]] : !fir.ref<!fir.box<!fir.ptr<f32>>>, !fir.ref<!fir.box<!fir.ptr<f32>>>) {
 433 ! CHECK:        acc.loop {{.*}} {
 434 ! CHECK:          acc.yield
 435 ! CHECK-NEXT:   }{{$}}
 436 ! CHECK:        acc.terminator
 437 ! CHECK-NEXT: }{{$}}
 438
 439   !$acc kernels loop seq
 440   DO i = 1, n
 441     a(i) = b(i)
 442   END DO
 443
 444 ! CHECK:      acc.kernels {{.*}} {
 445 ! CHECK:        acc.loop {{.*}} {
 446 ! CHECK:          acc.yield
 447 ! CHECK-NEXT:   } attributes {inclusiveUpperbound = array<i1: true>, seq = [#acc.device_type<none>]}
 448 ! CHECK:        acc.terminator
 449 ! CHECK-NEXT: }{{$}}
 450
 451   !$acc kernels loop auto
 452   DO i = 1, n
 453     a(i) = b(i)
 454   END DO
 455
 456 ! CHECK:      acc.kernels {{.*}} {
 457 ! CHECK:        acc.loop {{.*}} {
 458 ! CHECK:          acc.yield
 459 ! CHECK-NEXT:   } attributes {auto_ = [#acc.device_type<none>], inclusiveUpperbound = array<i1: true>}
 460 ! CHECK:        acc.terminator
 461 ! CHECK-NEXT: }{{$}}
 462
 463   !$acc kernels loop independent
 464   DO i = 1, n
 465     a(i) = b(i)
 466   END DO
 467
 468 ! CHECK:      acc.kernels {{.*}} {
 469 ! CHECK:        acc.loop {{.*}} {
 470 ! CHECK:          acc.yield
 471 ! CHECK-NEXT:   } attributes {inclusiveUpperbound = array<i1: true>, independent = [#acc.device_type<none>]}
 472 ! CHECK:        acc.terminator
 473 ! CHECK-NEXT: }{{$}}
 474
 475   !$acc kernels loop gang
 476   DO i = 1, n
 477     a(i) = b(i)
 478   END DO
 479
 480 ! CHECK:      acc.kernels {{.*}} {
 481 ! CHECK:        acc.loop {{.*}} gang {{.*}} {
 482 ! CHECK:          acc.yield
 483 ! CHECK-NEXT:   } attributes {inclusiveUpperbound = array<i1: true>}{{$}}
 484 ! CHECK:        acc.terminator
 485 ! CHECK-NEXT: }{{$}}
 486
 487   !$acc kernels loop gang(num: 8)
 488   DO i = 1, n
 489     a(i) = b(i)
 490   END DO
 491
 492 ! CHECK:      acc.kernels {{.*}} {
 493 ! CHECK:        [[GANGNUM1:%.*]] = arith.constant 8 : i32
 494 ! CHECK-NEXT:   acc.loop {{.*}} gang({num=[[GANGNUM1]] : i32}) {{.*}} {
 495 ! CHECK:          acc.yield
 496 ! CHECK-NEXT:   }{{$}}
 497 ! CHECK:        acc.terminator
 498 ! CHECK-NEXT: }{{$}}
 499
 500   !$acc kernels loop gang(num: gangNum)
 501   DO i = 1, n
 502     a(i) = b(i)
 503   END DO
 504
 505 ! CHECK:      acc.kernels {{.*}} {
 506 ! CHECK:        [[GANGNUM2:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 507 ! CHECK-NEXT:   acc.loop {{.*}} gang({num=[[GANGNUM2]] : i32}) {{.*}} {
 508 ! CHECK:          acc.yield
 509 ! CHECK-NEXT:   }{{$}}
 510 ! CHECK:        acc.terminator
 511 ! CHECK-NEXT: }{{$}}
 512
 513  !$acc kernels loop gang(num: gangNum, static: gangStatic)
 514   DO i = 1, n
 515     a(i) = b(i)
 516   END DO
 517
 518 ! CHECK:      acc.kernels {{.*}} {
 519 ! CHECK:        acc.loop {{.*}} gang({num=%{{.*}} : i32, static=%{{.*}} : i32})
 520 ! CHECK:          acc.yield
 521 ! CHECK-NEXT:   }{{$}}
 522 ! CHECK:        acc.terminator
 523 ! CHECK-NEXT: }{{$}}
 524
 525   !$acc kernels loop vector
 526   DO i = 1, n
 527     a(i) = b(i)
 528   END DO
 529
 530 ! CHECK:      acc.kernels {{.*}} {
 531 ! CHECK:        acc.loop {{.*}} vector {{.*}} {
 532 ! CHECK:          acc.yield
 533 ! CHECK-NEXT:   } attributes {inclusiveUpperbound = array<i1: true>}{{$}}
 534 ! CHECK:        acc.terminator
 535 ! CHECK-NEXT: }{{$}}
 536
 537   !$acc kernels loop vector(128)
 538   DO i = 1, n
 539     a(i) = b(i)
 540   END DO
 541
 542 ! CHECK:      acc.kernels {{.*}} {
 543 ! CHECK:        [[CONSTANT128:%.*]] = arith.constant 128 : i32
 544 ! CHECK:        acc.loop {{.*}} vector([[CONSTANT128]] : i32) {{.*}} {
 545 ! CHECK:          acc.yield
 546 ! CHECK-NEXT:   }{{$}}
 547 ! CHECK:        acc.terminator
 548 ! CHECK-NEXT: }{{$}}
 549
 550   !$acc kernels loop vector(vectorLength)
 551   DO i = 1, n
 552     a(i) = b(i)
 553   END DO
 554
 555 ! CHECK:      acc.kernels {{.*}} {
 556 ! CHECK:        [[VECTORLENGTH:%.*]] = fir.load %{{.*}} : !fir.ref<i32>
 557 ! CHECK:        acc.loop {{.*}} vector([[VECTORLENGTH]] : i32) {{.*}} {
 558 ! CHECK:          acc.yield
 559 ! CHECK-NEXT:   }{{$}}
 560 ! CHECK:        acc.terminator
 561 ! CHECK-NEXT: }{{$}}
 562
 563   !$acc kernels loop worker
 564   DO i = 1, n
 565     a(i) = b(i)
 566   END DO
 567
 568 ! CHECK:      acc.kernels {{.*}} {
 569 ! CHECK:        acc.loop {{.*}} worker {{.*}} {
 570 ! CHECK:          acc.yield
 571 ! CHECK-NEXT:   } attributes {inclusiveUpperbound = array<i1: true>}{{$}}
 572 ! CHECK:        acc.terminator
 573 ! CHECK-NEXT: }{{$}}
 574
 575   !$acc kernels loop worker(128)
 576   DO i = 1, n
 577     a(i) = b(i)
 578   END DO
 579
 580 ! CHECK:      acc.kernels {{.*}} {
 581 ! CHECK:        [[WORKER128:%.*]] = arith.constant 128 : i32
 582 ! CHECK:        acc.loop {{.*}} worker([[WORKER128]] : i32) {{.*}} {
 583 ! CHECK:          acc.yield
 584 ! CHECK-NEXT:   }{{$}}
 585 ! CHECK:        acc.terminator
 586 ! CHECK-NEXT: }{{$}}
 587
 588   !$acc kernels loop collapse(2)
 589   DO i = 1, n
 590     DO j = 1, n
 591       d(i, j) = e(i, j)
 592     END DO
 593   END DO
 594
 595 ! CHECK:      acc.kernels {{.*}} {
 596 ! CHECK:        acc.loop {{.*}} {
 597 ! CHECK:          acc.yield
 598 ! CHECK-NEXT:   } attributes {collapse = [2], collapseDeviceType = [#acc.device_type<none>], inclusiveUpperbound = array<i1: true, true>}
 599 ! CHECK:        acc.terminator
 600 ! CHECK-NEXT: }{{$}}
 601
 602   !$acc kernels loop
 603   DO i = 1, n
 604     !$acc loop
 605     DO j = 1, n
 606       d(i, j) = e(i, j)
 607     END DO
 608   END DO
 609
 610 ! CHECK:      acc.kernels {{.*}} {
 611 ! CHECK:        acc.loop {{.*}} {
 612 ! CHECK:            acc.loop {{.*}} {
 613 ! CHECK:              acc.yield
 614 ! CHECK-NEXT:     }{{$}}
 615 ! CHECK:          acc.yield
 616 ! CHECK-NEXT:   }{{$}}
 617 ! CHECK:        acc.terminator
 618 ! CHECK-NEXT: }{{$}}
 619
 620  !$acc kernels loop tile(2)
 621   DO i = 1, n
 622     a(i) = b(i)
 623   END DO
 624
 625 ! CHECK:      acc.kernels {{.*}} {
 626 ! CHECK:        [[TILESIZE:%.*]] = arith.constant 2 : i32
 627 ! CHECK:        acc.loop {{.*}} tile({[[TILESIZE]] : i32}) {{.*}} {
 628 ! CHECK:          acc.yield
 629 ! CHECK-NEXT:   }{{$}}
 630 ! CHECK:        acc.terminator
 631 ! CHECK-NEXT: }{{$}}
 632
 633  !$acc kernels loop tile(*)
 634   DO i = 1, n
 635     a(i) = b(i)
 636   END DO
 637
 638 ! CHECK:      acc.kernels {{.*}} {
 639 ! CHECK:        [[TILESIZEM1:%.*]] = arith.constant -1 : i32
 640 ! CHECK:        acc.loop {{.*}} tile({[[TILESIZEM1]] : i32}) {{.*}} {
 641 ! CHECK:          acc.yield
 642 ! CHECK-NEXT:   }{{$}}
 643 ! CHECK:        acc.terminator
 644 ! CHECK-NEXT: }{{$}}
 645
 646   !$acc kernels loop tile(2, 2)
 647   DO i = 1, n
 648     DO j = 1, n
 649       d(i, j) = e(i, j)
 650     END DO
 651   END DO
 652
 653 ! CHECK:      acc.kernels {{.*}} {
 654 ! CHECK:        [[TILESIZE1:%.*]] = arith.constant 2 : i32
 655 ! CHECK:        [[TILESIZE2:%.*]] = arith.constant 2 : i32
 656 ! CHECK:        acc.loop {{.*}} tile({[[TILESIZE1]] : i32, [[TILESIZE2]] : i32}) {{.*}} {
 657 ! CHECK:          acc.yield
 658 ! CHECK-NEXT:   }{{$}}
 659 ! CHECK:        acc.terminator
 660 ! CHECK-NEXT: }{{$}}
 661
 662   !$acc kernels loop tile(tileSize)
 663   DO i = 1, n
 664     a(i) = b(i)
 665   END DO
 666
 667 ! CHECK:      acc.kernels {{.*}} {
 668 ! CHECK:        acc.loop {{.*}} tile({%{{.*}} : i32}) {{.*}} {
 669 ! CHECK:          acc.yield
 670 ! CHECK-NEXT:   }{{$}}
 671 ! CHECK:        acc.terminator
 672 ! CHECK-NEXT: }{{$}}
 673
 674   !$acc kernels loop tile(tileSize, tileSize)
 675   DO i = 1, n
 676     DO j = 1, n
 677       d(i, j) = e(i, j)
 678     END DO
 679   END DO
 680
 681 ! CHECK:      acc.kernels {{.*}} {
 682 ! CHECK:        acc.loop {{.*}} tile({%{{.*}} : i32, %{{.*}} : i32}) {{.*}} {
 683 ! CHECK:          acc.yield
 684 ! CHECK-NEXT:   }{{$}}
 685 ! CHECK:        acc.terminator
 686 ! CHECK-NEXT: }{{$}}
 687
 688   !$acc kernels loop reduction(+:reduction_r) reduction(*:reduction_i)
 689   do i = 1, n
 690     reduction_r = reduction_r + a(i)
 691     reduction_i = 1
 692   end do
 693
 694 ! CHECK:      %[[COPYINREDR:.*]] = acc.copyin varPtr(%{{.*}} : !fir.ref<f32>) -> !fir.ref<f32> {dataClause = #acc<data_clause acc_reduction>, implicit = true, name = "reduction_r"}
 695 ! CHECK:      %[[COPYINREDI:.*]] = acc.copyin varPtr(%{{.*}} : !fir.ref<i32>) -> !fir.ref<i32> {dataClause = #acc<data_clause acc_reduction>, implicit = true, name = "reduction_i"}
 696 ! CHECK:      acc.kernels {{.*}} dataOperands(%[[COPYINREDR]], %[[COPYINREDI]] : !fir.ref<f32>, !fir.ref<i32>) {
 697 ! CHECK:        acc.loop {{.*}} reduction(@reduction_add_ref_f32 -> %{{.*}} : !fir.ref<f32>, @reduction_mul_ref_i32 -> %{{.*}} : !fir.ref<i32>) {{.*}} {
 698 ! CHECK:          acc.yield
 699 ! CHECK-NEXT:   }{{$}}
 700 ! CHECK:        acc.terminator
 701 ! CHECK-NEXT: }{{$}}
 702 ! CHECK:      acc.copyout accPtr(%[[COPYINREDR]] : !fir.ref<f32>) to varPtr(%{{.*}} : !fir.ref<f32>) {dataClause = #acc<data_clause acc_reduction>, implicit = true, name = "reduction_r"}
 703 ! CHECK:      acc.copyout accPtr(%[[COPYINREDI]] : !fir.ref<i32>) to varPtr(%{{.*}} : !fir.ref<i32>) {dataClause = #acc<data_clause acc_reduction>, implicit = true, name = "reduction_i"}
 704
 705 end subroutine