llvm/test/CodeGen/AMDGPU/llvm.amdgcn.wavefrontsize.ll

   1 ; RUN: llc -march=amdgcn -mcpu=fiji -verify-machineinstrs < %s | FileCheck -check-prefixes=GCN,W64 %s
   2 ; RUN: llc -march=amdgcn -mcpu=gfx1010 -mattr=+wavefrontsize32,-wavefrontsize64 -verify-machineinstrs < %s | FileCheck -check-prefixes=GCN,W32 %s
   3 ; RUN: llc -march=amdgcn -mcpu=gfx1010 -mattr=-wavefrontsize32,+wavefrontsize64 -verify-machineinstrs < %s | FileCheck -check-prefixes=GCN,W64 %s
   4 ; RUN: llc -march=amdgcn -mcpu=gfx1100 -mattr=+wavefrontsize32,-wavefrontsize64 -verify-machineinstrs -amdgpu-enable-vopd=0 < %s | FileCheck -check-prefixes=GCN,W32 %s
   5 ; RUN: llc -march=amdgcn -mcpu=gfx1100 -mattr=-wavefrontsize32,+wavefrontsize64 -verify-machineinstrs < %s | FileCheck -check-prefixes=GCN,W64 %s
   6
   7 ; RUN: opt -O3 -S < %s | FileCheck -check-prefix=OPT %s
   8 ; RUN: opt -mtriple=amdgcn-- -O3 -S < %s | FileCheck -check-prefix=OPT %s
   9 ; RUN: opt -mtriple=amdgcn-- -O3 -mattr=+wavefrontsize32 -S < %s | FileCheck -check-prefix=OPT %s
  10 ; RUN: opt -mtriple=amdgcn-- -passes='default<O3>' -mattr=+wavefrontsize32 -S < %s | FileCheck -check-prefix=OPT %s
  11 ; RUN: opt -mtriple=amdgcn-- -O3 -mattr=+wavefrontsize64 -S < %s | FileCheck -check-prefix=OPT %s
  12 ; RUN: opt -mtriple=amdgcn-- -mcpu=tonga -O3 -S < %s | FileCheck -check-prefix=OPT %s
  13 ; RUN: opt -mtriple=amdgcn-- -mcpu=gfx1010 -O3 -mattr=+wavefrontsize32,-wavefrontsize64 -S < %s | FileCheck -check-prefix=OPT %s
  14 ; RUN: opt -mtriple=amdgcn-- -mcpu=gfx1010 -O3 -mattr=-wavefrontsize32,+wavefrontsize64 -S < %s | FileCheck -check-prefix=OPT %s
  15 ; RUN: opt -mtriple=amdgcn-- -mcpu=gfx1100 -O3 -mattr=+wavefrontsize32,-wavefrontsize64 -S < %s | FileCheck -check-prefix=OPT %s
  16 ; RUN: opt -mtriple=amdgcn-- -mcpu=gfx1100 -O3 -mattr=-wavefrontsize32,+wavefrontsize64 -S < %s | FileCheck -check-prefix=OPT %s
  17
  18 ; GCN-LABEL: {{^}}fold_wavefrontsize:
  19 ; OPT-LABEL: define amdgpu_kernel void @fold_wavefrontsize(
  20
  21 ; W32:       v_mov_b32_e32 [[V:v[0-9]+]], 32
  22 ; W64:       v_mov_b32_e32 [[V:v[0-9]+]], 64
  23 ; GCN:       store_{{dword|b32}} v{{.+}}, [[V]]
  24
  25 ; OPT:   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize()
  26 ; OPT:   store i32 %tmp, ptr addrspace(1) %arg, align 4
  27 ; OPT-NEXT:  ret void
  28
  29 define amdgpu_kernel void @fold_wavefrontsize(ptr addrspace(1) nocapture %arg) {
  30 bb:
  31   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize() #0
  32   store i32 %tmp, ptr addrspace(1) %arg, align 4
  33   ret void
  34 }
  35
  36 ; GCN-LABEL: {{^}}fold_and_optimize_wavefrontsize:
  37 ; OPT-LABEL: define amdgpu_kernel void @fold_and_optimize_wavefrontsize(
  38
  39 ; W32:       v_mov_b32_e32 [[V:v[0-9]+]], 1{{$}}
  40 ; W64:       v_mov_b32_e32 [[V:v[0-9]+]], 2{{$}}
  41 ; GCN-NOT:   cndmask
  42 ; GCN:       store_{{dword|b32}} v{{.+}}, [[V]]
  43
  44 ; OPT:   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize()
  45 ; OPT:   %tmp1 = icmp ugt i32 %tmp, 32
  46 ; OPT:   %tmp2 = select i1 %tmp1, i32 2, i32 1
  47 ; OPT:   store i32 %tmp2, ptr addrspace(1) %arg
  48 ; OPT-NEXT:  ret void
  49
  50 define amdgpu_kernel void @fold_and_optimize_wavefrontsize(ptr addrspace(1) nocapture %arg) {
  51 bb:
  52   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize() #0
  53   %tmp1 = icmp ugt i32 %tmp, 32
  54   %tmp2 = select i1 %tmp1, i32 2, i32 1
  55   store i32 %tmp2, ptr addrspace(1) %arg
  56   ret void
  57 }
  58
  59 ; GCN-LABEL: {{^}}fold_and_optimize_if_wavefrontsize:
  60 ; OPT-LABEL: define amdgpu_kernel void @fold_and_optimize_if_wavefrontsize(
  61
  62 ; OPT:       bb:
  63 ; OPT:   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize()
  64 ; OPT:   %tmp1 = icmp ugt i32 %tmp, 32
  65 ; OPT:   bb3:
  66 ; OPT-NEXT:  ret void
  67
  68 define amdgpu_kernel void @fold_and_optimize_if_wavefrontsize(ptr addrspace(1) nocapture %arg) {
  69 bb:
  70   %tmp = tail call i32 @llvm.amdgcn.wavefrontsize() #0
  71   %tmp1 = icmp ugt i32 %tmp, 32
  72   br i1 %tmp1, label %bb2, label %bb3
  73
  74 bb2:                                              ; preds = %bb
  75   store i32 1, ptr addrspace(1) %arg, align 4
  76   br label %bb3
  77
  78 bb3:                                              ; preds = %bb2, %bb
  79   ret void
  80 }
  81
  82 declare i32 @llvm.amdgcn.wavefrontsize() #0
  83
  84 attributes #0 = { nounwind readnone speculatable }