xref: /llvm-project/clang/test/CodeGenOpenCL/arm-integer-dot-product.cl (revision b0e4897a1bd2ea5eeeb8c104e32df1402a397ac1)
120d93267SSven van Haastregt// RUN: %clang_cc1 %s -triple spir-unknown-unknown -finclude-default-header -fdeclare-opencl-builtins -cl-std=CL1.2 -emit-llvm -o - -O0 | FileCheck %s
2aa7754ccSKevin Petit
358d18ddeSAnastasia Stulova// Pragmas are only accepted for backward compatibility.
4*b0e4897aSSven van Haastregt// The builtins are made available with the following defines.
5*b0e4897aSSven van Haastregt
6*b0e4897aSSven van Haastregt#define cl_arm_integer_dot_product_int8 1
7*b0e4897aSSven van Haastregt#define cl_arm_integer_dot_product_accumulate_int8 1
8*b0e4897aSSven van Haastregt#define cl_arm_integer_dot_product_accumulate_int16 1
9*b0e4897aSSven van Haastregt#define cl_arm_integer_dot_product_accumulate_saturate_int8 1
1058d18ddeSAnastasia Stulova
11aa7754ccSKevin Petit#pragma OPENCL EXTENSION cl_arm_integer_dot_product_int8 : enable
1258d18ddeSAnastasia Stulova#pragma OPENCL EXTENSION cl_arm_integer_dot_product_int8 : disable
13aa7754ccSKevin Petitvoid test_int8(uchar4 ua, uchar4 ub, char4 sa, char4 sb) {
14aa7754ccSKevin Petit    uint ur = arm_dot(ua, ub);
15aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z7arm_dotDv4_hS_
16aa7754ccSKevin Petit    int sr = arm_dot(sa, sb);
17aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z7arm_dotDv4_cS_
18aa7754ccSKevin Petit}
19aa7754ccSKevin Petit
20aa7754ccSKevin Petit#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_int8 : enable
2158d18ddeSAnastasia Stulova#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_int8 : disable
22aa7754ccSKevin Petitvoid test_accumulate_int8(uchar4 ua, uchar4 ub, uint uc, char4 sa, char4 sb, int c) {
23aa7754ccSKevin Petit    uint ur = arm_dot_acc(ua, ub, uc);
24aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z11arm_dot_accDv4_hS_j
25aa7754ccSKevin Petit    int sr = arm_dot_acc(sa, sb, c);
26aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z11arm_dot_accDv4_cS_i
27aa7754ccSKevin Petit}
28aa7754ccSKevin Petit
29aa7754ccSKevin Petit#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_int16 : enable
3058d18ddeSAnastasia Stulova#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_int16 : disable
31aa7754ccSKevin Petitvoid test_accumulate_int16(ushort2 ua, ushort2 ub, uint uc, short2 sa, short2 sb, int c) {
32aa7754ccSKevin Petit    uint ur = arm_dot_acc(ua, ub, uc);
33aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z11arm_dot_accDv2_tS_j
34aa7754ccSKevin Petit    int sr = arm_dot_acc(sa, sb, c);
35aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z11arm_dot_accDv2_sS_i
36aa7754ccSKevin Petit}
37aa7754ccSKevin Petit
38aa7754ccSKevin Petit#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_saturate_int8 : enable
3958d18ddeSAnastasia Stulova#pragma OPENCL EXTENSION cl_arm_integer_dot_product_accumulate_saturate_int8 : disable
40aa7754ccSKevin Petitvoid test_accumulate_saturate_int8(uchar4 ua, uchar4 ub, uint uc, char4 sa, char4 sb, int c) {
41aa7754ccSKevin Petit    uint ur = arm_dot_acc_sat(ua, ub, uc);
42aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z15arm_dot_acc_satDv4_hS_j
43aa7754ccSKevin Petit    int sr = arm_dot_acc_sat(sa, sb, c);
44aa7754ccSKevin Petit    // CHECK: call spir_func i32 @_Z15arm_dot_acc_satDv4_cS_i
45aa7754ccSKevin Petit}
46