clang/test/CodeGenOpenCL/preserve_vec3.cl

   1 // RUN: %clang_cc1 -no-opaque-pointers %s -emit-llvm -o - -triple spir-unknown-unknown -fpreserve-vec3-type | FileCheck %s
   2
   3 typedef char char3 __attribute__((ext_vector_type(3)));
   4 typedef char char8 __attribute__((ext_vector_type(8)));
   5 typedef short short3 __attribute__((ext_vector_type(3)));
   6 typedef double double2 __attribute__((ext_vector_type(2)));
   7 typedef float float3 __attribute__((ext_vector_type(3)));
   8 typedef float float4 __attribute__((ext_vector_type(4)));
   9
  10 void kernel foo(global float3 *a, global float3 *b) {
  11   // CHECK-LABEL: spir_kernel void @foo
  12   // CHECK: %[[LOAD_A:.*]] = load <3 x float>, <3 x float> addrspace(1)* %a
  13   // CHECK: store <3 x float> %[[LOAD_A]], <3 x float> addrspace(1)* %b
  14   *b = *a;
  15 }
  16
  17 void kernel float4_to_float3(global float3 *a, global float4 *b) {
  18   // CHECK-LABEL: spir_kernel void @float4_to_float3
  19   // CHECK: %[[LOAD_A:.*]] = load <4 x float>, <4 x float> addrspace(1)* %b, align 16
  20   // CHECK: %[[ASTYPE:.*]] = shufflevector <4 x float> %[[LOAD_A]], <4 x float> poison, <3 x i32> <i32 0, i32 1, i32 2>
  21   // CHECK: store <3 x float> %[[ASTYPE]], <3 x float> addrspace(1)* %a, align 16
  22   *a = __builtin_astype(*b, float3);
  23 }
  24
  25 void kernel float3_to_float4(global float3 *a, global float4 *b) {
  26   // CHECK-LABEL: spir_kernel void @float3_to_float4
  27   // CHECK: %[[LOAD_A:.*]] = load <3 x float>, <3 x float> addrspace(1)* %a, align 16
  28   // CHECK: %[[ASTYPE:.*]] = shufflevector <3 x float> %[[LOAD_A]], <3 x float> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 undef>
  29   // CHECK: store <4 x float> %[[ASTYPE]], <4 x float> addrspace(1)* %b, align 16
  30   *b = __builtin_astype(*a, float4);
  31 }
  32
  33 void kernel float3_to_double2(global float3 *a, global double2 *b) {
  34   // CHECK-LABEL: spir_kernel void @float3_to_double2
  35   // CHECK: %[[LOAD_A:.*]] = load <3 x float>, <3 x float> addrspace(1)* %a, align 16
  36   // CHECK: %[[ASTYPE:.*]] = shufflevector <3 x float> %[[LOAD_A]], <3 x float> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 undef>
  37   // CHECK: %[[OUT_BC:.*]] = bitcast <2 x double> addrspace(1)* %b to <4 x float> addrspace(1)*
  38   // CHECK: store <4 x float> %[[ASTYPE]], <4 x float> addrspace(1)* %[[OUT_BC]], align 16
  39   *b = __builtin_astype(*a, double2);
  40 }
  41
  42 void kernel char8_to_short3(global short3 *a, global char8 *b) {
  43   // CHECK-LABEL: spir_kernel void @char8_to_short3
  44   // CHECK: %[[IN_BC:.*]] = bitcast <8 x i8> addrspace(1)* %b to <4 x i16> addrspace(1)*
  45   // CHECK: %[[LOAD_B:.*]] = load <4 x i16>, <4 x i16> addrspace(1)* %[[IN_BC]]
  46   // CHECK: %[[ASTYPE:.*]] = shufflevector <4 x i16> %[[LOAD_B]], <4 x i16> poison, <3 x i32> <i32 0, i32 1, i32 2>
  47   // CHECK: store <3 x i16> %[[ASTYPE]], <3 x i16> addrspace(1)* %a, align 8
  48   *a = __builtin_astype(*b, short3);
  49 }
  50
  51 void from_char3(char3 a, global int *out) {
  52   // CHECK-LABEL: void @from_char3
  53   // CHECK: %[[ASTYPE:.*]] = shufflevector <3 x i8> %a, <3 x i8> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 undef>
  54   // CHECK: %[[OUT_BC:.*]] = bitcast i32 addrspace(1)* %out to <4 x i8> addrspace(1)*
  55   // CHECK: store <4 x i8> %[[ASTYPE]], <4 x i8> addrspace(1)* %[[OUT_BC]]
  56   *out = __builtin_astype(a, int);
  57 }
  58
  59 void from_short3(short3 a, global long *out) {
  60   // CHECK-LABEL: void @from_short3
  61   // CHECK: %[[ASTYPE:.*]] = shufflevector <3 x i16> %a, <3 x i16> poison, <4 x i32> <i32 0, i32 1, i32 2, i32 undef>
  62   // CHECK: %[[OUT_BC:.*]] = bitcast i64 addrspace(1)* %out to <4 x i16> addrspace(1)*
  63   // CHECK: store <4 x i16> %[[ASTYPE]], <4 x i16> addrspace(1)* %[[OUT_BC]]
  64   *out = __builtin_astype(a, long);
  65 }
  66
  67 void scalar_to_char3(int a, global char3 *out) {
  68   // CHECK-LABEL: void @scalar_to_char3
  69   // CHECK: %[[IN_BC:.*]] = bitcast i32 %a to <4 x i8>
  70   // CHECK: %[[ASTYPE:.*]] = shufflevector <4 x i8> %[[IN_BC]], <4 x i8> poison, <3 x i32> <i32 0, i32 1, i32 2>
  71   // CHECK: store <3 x i8> %[[ASTYPE]], <3 x i8> addrspace(1)* %out
  72   *out = __builtin_astype(a, char3);
  73 }
  74
  75 void scalar_to_short3(long a, global short3 *out) {
  76   // CHECK-LABEL: void @scalar_to_short3
  77   // CHECK: %[[IN_BC:.*]] = bitcast i64 %a to <4 x i16>
  78   // CHECK: %[[ASTYPE:.*]] = shufflevector <4 x i16> %[[IN_BC]], <4 x i16> poison, <3 x i32> <i32 0, i32 1, i32 2>
  79   // CHECK: store <3 x i16> %[[ASTYPE]], <3 x i16> addrspace(1)* %out
  80   *out = __builtin_astype(a, short3);
  81 }