clang/test/CodeGenOpenCL/builtins-amdgcn-dl-insts-gfx11.cl

   1 // REQUIRES: amdgpu-registered-target
   2
   3 // RUN: %clang_cc1 -triple amdgcn-unknown-unknown -target-cpu gfx1100 -S -emit-llvm -o - %s | FileCheck %s
   4
   5 typedef unsigned int uint;
   6 typedef half __attribute__((ext_vector_type(2))) half2;
   7 typedef short __attribute__((ext_vector_type(2))) short2;
   8 typedef unsigned short __attribute__((ext_vector_type(2))) ushort2;
   9
  10 // CHECK-LABEL: @builtins_amdgcn_dl_insts
  11 // CHECK: call float @llvm.amdgcn.fdot2(<2 x half> %v2hA, <2 x half> %v2hB, float %fC, i1 false)
  12 // CHECK: call float @llvm.amdgcn.fdot2(<2 x half> %v2hA, <2 x half> %v2hB, float %fC, i1 true)
  13 // CHECK: call half @llvm.amdgcn.fdot2.f16.f16(<2 x half> %v2hA, <2 x half> %v2hB, half %hC)
  14 // CHECK: call i16 @llvm.amdgcn.fdot2.bf16.bf16(<2 x i16> %v2ssA, <2 x i16> %v2ssB, i16 %sC)
  15 // CHECK: call float @llvm.amdgcn.fdot2.f32.bf16(<2 x i16> %v2ssA, <2 x i16> %v2ssB, float %fC, i1 false)
  16 // CHECK: call float @llvm.amdgcn.fdot2.f32.bf16(<2 x i16> %v2ssA, <2 x i16> %v2ssB, float %fC, i1 true)
  17 // CHECK: call i32 @llvm.amdgcn.sdot4(i32 %siA, i32 %siB, i32 %siC, i1 false)
  18 // CHECK: call i32 @llvm.amdgcn.sdot4(i32 %siA, i32 %siB, i32 %siC, i1 true)
  19 // CHECK: call i32 @llvm.amdgcn.udot4(i32 %uiA, i32 %uiB, i32 %uiC, i1 false)
  20 // CHECK: call i32 @llvm.amdgcn.udot4(i32 %uiA, i32 %uiB, i32 %uiC, i1 true)
  21 // CHECK: call i32 @llvm.amdgcn.sudot4(i1 true, i32 %A, i1 false, i32 %B, i32 %C, i1 false)
  22 // CHECK: call i32 @llvm.amdgcn.sudot4(i1 false, i32 %A, i1 true, i32 %B, i32 %C, i1 true)
  23 // CHECK: call i32 @llvm.amdgcn.sdot8(i32 %siA, i32 %siB, i32 %siC, i1 false)
  24 // CHECK: call i32 @llvm.amdgcn.sdot8(i32 %siA, i32 %siB, i32 %siC, i1 true)
  25 // CHECK: call i32 @llvm.amdgcn.udot8(i32 %uiA, i32 %uiB, i32 %uiC, i1 false)
  26 // CHECK: call i32 @llvm.amdgcn.udot8(i32 %uiA, i32 %uiB, i32 %uiC, i1 true)
  27 // CHECK: call i32 @llvm.amdgcn.sudot8(i1 false, i32 %A, i1 true, i32 %B, i32 %C, i1 false)
  28 // CHECK: call i32 @llvm.amdgcn.sudot8(i1 true, i32 %A, i1 false, i32 %B, i32 %C, i1 true)
  29 #pragma OPENCL EXTENSION cl_khr_fp16 : enable
  30 kernel void builtins_amdgcn_dl_insts_err(
  31     global float *fOut, global int *siOut, global uint *uiOut,
  32     global short *sOut, global int *iOut, global half *hOut,
  33     half2 v2hA, half2 v2hB, float fC, half hC,
  34     short2 v2ssA, short2 v2ssB, short sC, int siA, int siB, int siC,
  35     ushort2 v2usA, ushort2 v2usB, uint uiA, uint uiB, uint uiC,
  36     int A, int B, int C) {
  37   fOut[0] = __builtin_amdgcn_fdot2(v2hA, v2hB, fC, false);
  38   fOut[1] = __builtin_amdgcn_fdot2(v2hA, v2hB, fC, true);
  39
  40   hOut[0] = __builtin_amdgcn_fdot2_f16_f16(v2hA, v2hB, hC);
  41
  42   sOut[0] = __builtin_amdgcn_fdot2_bf16_bf16(v2ssA, v2ssB, sC);
  43
  44   fOut[3] = __builtin_amdgcn_fdot2_f32_bf16(v2ssA, v2ssB, fC, false);
  45   fOut[4] = __builtin_amdgcn_fdot2_f32_bf16(v2ssA, v2ssB, fC, true);
  46
  47   siOut[2] = __builtin_amdgcn_sdot4(siA, siB, siC, false);
  48   siOut[3] = __builtin_amdgcn_sdot4(siA, siB, siC, true);
  49
  50   uiOut[2] = __builtin_amdgcn_udot4(uiA, uiB, uiC, false);
  51   uiOut[3] = __builtin_amdgcn_udot4(uiA, uiB, uiC, true);
  52
  53   iOut[0] = __builtin_amdgcn_sudot4(true, A, false, B, C, false);
  54   iOut[1] = __builtin_amdgcn_sudot4(false, A, true, B, C, true);
  55
  56   siOut[4] = __builtin_amdgcn_sdot8(siA, siB, siC, false);
  57   siOut[5] = __builtin_amdgcn_sdot8(siA, siB, siC, true);
  58
  59   uiOut[4] = __builtin_amdgcn_udot8(uiA, uiB, uiC, false);
  60   uiOut[5] = __builtin_amdgcn_udot8(uiA, uiB, uiC, true);
  61
  62   iOut[3] = __builtin_amdgcn_sudot8(false, A, true, B, C, false);
  63   iOut[4] = __builtin_amdgcn_sudot8(true, A, false, B, C, true);
  64 }