llvm/test/CodeGen/SPIRV/pointers/complex.ll

   1 ; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s
   2 ; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
   3
   4 ; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s
   5 ; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown %s -o - -filetype=obj | spirv-val %}
   6
   7 ; CHECK-DAG: OpName %[[#Foo:]] "foo"
   8 ; CHECK-DAG: OpName %[[#CSQRT:]] "csqrt"
   9 ; CHECK-DAG: %[[#FloatTy:]] = OpTypeFloat 64
  10 ; CHECK-DAG: %[[#StructTy:]] = OpTypeStruct %[[#FloatTy]] %[[#FloatTy]]
  11 ; CHECK-DAG: %[[#GPtrStructTy:]] = OpTypePointer Generic %[[#StructTy]]
  12 ; CHECK-DAG: %[[#FPtrStructTy:]] = OpTypePointer Function %[[#StructTy]]
  13 ; CHECK-DAG: %[[#WrapTy:]] = OpTypeStruct %[[#StructTy]]
  14 ; CHECK-DAG: %[[#Int64Ty:]] = OpTypeInt 64 0
  15 ; CHECK-DAG: %[[#Int32Ty:]] = OpTypeInt 32 0
  16 ; CHECK-DAG: %[[#C1:]] = OpConstant %[[#Int32Ty]] 1
  17 ; CHECK-DAG: %[[#ArrayTy:]] = OpTypeArray %[[#Int64Ty]] %[[#C1]]
  18 ; CHECK-DAG: %[[#SArrayTy:]] = OpTypeStruct %[[#ArrayTy]]
  19 ; CHECK-DAG: %[[#SSArrayTy:]] = OpTypeStruct %[[#SArrayTy]]
  20 ; CHECK-DAG: %[[#CWPtrComplexTy:]] = OpTypePointer CrossWorkgroup %[[#WrapTy]]
  21 ; CHECK-DAG: %[[#IdTy:]] = OpTypePointer Function %[[#SSArrayTy]]
  22 ; CHECK: %[[#Foo]] = OpFunction
  23 ; CHECK: OpFunctionParameter %[[#CWPtrComplexTy]]
  24 ; CHECK: OpFunctionParameter %[[#IdTy]]
  25 ; CHECK: %[[#CSQRT]] = OpFunction
  26 ; CHECK: OpFunctionParameter %[[#GPtrStructTy]]
  27 ; CHECK: OpFunctionParameter %[[#FPtrStructTy]]
  28
  29 %"class.id" = type { %"class.array" }
  30 %"class.array" = type { [1 x i64] }
  31 %"class.complex" = type { { double, double } }
  32
  33 define weak_odr dso_local spir_kernel void @foo(ptr addrspace(1) align 8 %_arg_buf_out1_access, ptr byval(%"class.id") align 8 %_arg_buf_out1_access3) {
  34 entry:
  35   %tmp.i.i = alloca { double, double }, align 8
  36   %byval-temp.i.i = alloca { double, double }, align 8
  37   %idxvalue = load i64, ptr %_arg_buf_out1_access3, align 8
  38   %add.ptr.i = getelementptr inbounds %"class.complex", ptr addrspace(1) %_arg_buf_out1_access, i64 %idxvalue
  39   %tmp.ascast.i.i = addrspacecast ptr %tmp.i.i to ptr addrspace(4)
  40   %byval-temp.imagp.i.i = getelementptr inbounds i8, ptr %byval-temp.i.i, i64 8
  41   store double -1.000000e+00, ptr %byval-temp.i.i, align 8
  42   store double 0.000000e+00, ptr %byval-temp.imagp.i.i, align 8
  43   call spir_func void @csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8 %tmp.ascast.i.i, ptr nonnull byval({ double, double }) align 8 %byval-temp.i.i)
  44   %tmp.ascast.real.i.i = load double, ptr %tmp.i.i, align 8
  45   %tmp.ascast.imagp.i.i = getelementptr inbounds i8, ptr %tmp.i.i, i64 8
  46   %tmp.ascast.imag.i.i = load double, ptr %tmp.ascast.imagp.i.i, align 8
  47   store double %tmp.ascast.real.i.i, ptr addrspace(1) %add.ptr.i, align 8
  48   %dest = getelementptr inbounds i8, ptr addrspace(1) %add.ptr.i, i64 8
  49   store double %tmp.ascast.imag.i.i, ptr addrspace(1) %dest, align 8
  50   ret void
  51 }
  52
  53 define weak dso_local spir_func void @csqrt(ptr addrspace(4) dead_on_unwind noalias writable sret({ double, double }) align 8 %agg.result, ptr byval({ double, double }) align 8 %z) {
  54 entry:
  55   %tmp = alloca { double, double }, align 8
  56   %byval-temp = alloca { double, double }, align 8
  57   %tmp.ascast = addrspacecast ptr %tmp to ptr addrspace(4)
  58   %z.ascast.real = load double, ptr %z, align 8
  59   %z.ascast.imagp = getelementptr inbounds i8, ptr %z, i64 8
  60   %z.ascast.imag = load double, ptr %z.ascast.imagp, align 8
  61   %byval-temp.imagp = getelementptr inbounds i8, ptr %byval-temp, i64 8
  62   store double %z.ascast.real, ptr %byval-temp, align 8
  63   store double %z.ascast.imag, ptr %byval-temp.imagp, align 8
  64   call spir_func void @__devicelib_csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8 %tmp.ascast, ptr nonnull byval({ double, double }) align 8 %byval-temp) #7
  65   %tmp.ascast.real = load double, ptr %tmp, align 8
  66   %tmp.ascast.imagp = getelementptr inbounds i8, ptr %tmp, i64 8
  67   %tmp.ascast.imag = load double, ptr %tmp.ascast.imagp, align 8
  68   %agg.result.imagp = getelementptr inbounds i8, ptr addrspace(4) %agg.result, i64 8
  69   store double %tmp.ascast.real, ptr addrspace(4) %agg.result, align 8
  70   store double %tmp.ascast.imag, ptr addrspace(4) %agg.result.imagp, align 8
  71   ret void
  72 }
  73
  74 declare extern_weak dso_local spir_func void @__devicelib_csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8, ptr byval({ double, double }) align 8) local_unnamed_addr #3