test/CodeGen/AMDGPU/llvm.round.f64.ll

   1 ; RUN: llc -march=amdgcn < %s | FileCheck -check-prefix=SI -check-prefix=FUNC %s
   2
   3 ; FUNC-LABEL: {{^}}round_f64:
   4 ; SI: s_endpgm
   5 define amdgpu_kernel void @round_f64(double addrspace(1)* %out, double %x) #0 {
   6   %result = call double @llvm.round.f64(double %x) #1
   7   store double %result, double addrspace(1)* %out
   8   ret void
   9 }
  10
  11 ; This is a pretty large function, so just test a few of the
  12 ; instructions that are necessary.
  13
  14 ; FUNC-LABEL: {{^}}v_round_f64:
  15 ; SI: buffer_load_dwordx2
  16 ; SI-DAG: v_bfe_u32 [[EXP:v[0-9]+]], v{{[0-9]+}}, 20, 11
  17
  18 ; SI-DAG: v_not_b32_e32
  19 ; SI-DAG: v_not_b32_e32
  20
  21 ; SI-DAG: v_cmp_eq_u32
  22
  23 ; SI-DAG: s_brev_b32 [[BFIMASK:s[0-9]+]], -2{{$}}
  24 ; SI-DAG: v_cmp_gt_i32
  25 ; SI-DAG: v_bfi_b32 [[COPYSIGN:v[0-9]+]], [[BFIMASK]]
  26
  27 ; SI: buffer_store_dwordx2
  28 ; SI: s_endpgm
  29 define amdgpu_kernel void @v_round_f64(double addrspace(1)* %out, double addrspace(1)* %in) #0 {
  30   %tid = call i32 @llvm.amdgcn.workitem.id.x() #1
  31   %gep = getelementptr double, double addrspace(1)* %in, i32 %tid
  32   %out.gep = getelementptr double, double addrspace(1)* %out, i32 %tid
  33   %x = load double, double addrspace(1)* %gep
  34   %result = call double @llvm.round.f64(double %x) #1
  35   store double %result, double addrspace(1)* %out.gep
  36   ret void
  37 }
  38
  39 ; FUNC-LABEL: {{^}}round_v2f64:
  40 ; SI: s_endpgm
  41 define amdgpu_kernel void @round_v2f64(<2 x double> addrspace(1)* %out, <2 x double> %in) #0 {
  42   %result = call <2 x double> @llvm.round.v2f64(<2 x double> %in) #1
  43   store <2 x double> %result, <2 x double> addrspace(1)* %out
  44   ret void
  45 }
  46
  47 ; FUNC-LABEL: {{^}}round_v4f64:
  48 ; SI: s_endpgm
  49 define amdgpu_kernel void @round_v4f64(<4 x double> addrspace(1)* %out, <4 x double> %in) #0 {
  50   %result = call <4 x double> @llvm.round.v4f64(<4 x double> %in) #1
  51   store <4 x double> %result, <4 x double> addrspace(1)* %out
  52   ret void
  53 }
  54
  55 ; FUNC-LABEL: {{^}}round_v8f64:
  56 ; SI: s_endpgm
  57 define amdgpu_kernel void @round_v8f64(<8 x double> addrspace(1)* %out, <8 x double> %in) #0 {
  58   %result = call <8 x double> @llvm.round.v8f64(<8 x double> %in) #1
  59   store <8 x double> %result, <8 x double> addrspace(1)* %out
  60   ret void
  61 }
  62
  63 declare i32 @llvm.amdgcn.workitem.id.x() #1
  64
  65 declare double @llvm.round.f64(double) #1
  66 declare <2 x double> @llvm.round.v2f64(<2 x double>) #1
  67 declare <4 x double> @llvm.round.v4f64(<4 x double>) #1
  68 declare <8 x double> @llvm.round.v8f64(<8 x double>) #1
  69
  70 attributes #0 = { nounwind }
  71 attributes #1 = { nounwind readnone }