// REQUIRES: amdgpu-registered-target
// RUN: %clang_cc1 -triple amdgcn-unknown-unknown -target-cpu gfx1010 -emit-llvm -o - %s | FileCheck %s
// RUN: %clang_cc1 -triple amdgcn-unknown-unknown -target-cpu gfx1011 -emit-llvm -o - %s | FileCheck %s
// RUN: %clang_cc1 -triple amdgcn-unknown-unknown -target-cpu gfx1012 -emit-llvm -o - %s | FileCheck %s
// RUN: %clang_cc1 -triple spirv64-amd-amdhsa -emit-llvm -o - %s | FileCheck %s

#pragma OPENCL EXTENSION cl_khr_fp16 : enable

typedef unsigned int uint;
typedef unsigned long ulong;

// CHECK-LABEL: @test_permlane16(
// CHECK: {{.*}}call{{.*}} i32 @llvm.amdgcn.permlane16.i32(i32 %a, i32 %b, i32 %c, i32 %d, i1 false, i1 false)
void test_permlane16(global uint* out, uint a, uint b, uint c, uint d) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_permlane16))
    *out = __builtin_amdgcn_permlane16(a, b, c, d, 0, 0);
}

// CHECK-LABEL: @test_permlanex16(
// CHECK: {{.*}}call{{.*}} i32 @llvm.amdgcn.permlanex16.i32(i32 %a, i32 %b, i32 %c, i32 %d, i1 false, i1 false)
void test_permlanex16(global uint* out, uint a, uint b, uint c, uint d) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_permlanex16))
    *out = __builtin_amdgcn_permlanex16(a, b, c, d, 0, 0);
}

// CHECK-LABEL: @test_mov_dpp8_uint(
// CHECK:      {{.*}}call{{.*}} i32 @llvm.amdgcn.mov.dpp8.i32(i32 %a, i32 1)
// CHECK-NEXT: store i32 %[[#]],
void test_mov_dpp8_uint(global uint* out, uint a) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *out = __builtin_amdgcn_mov_dpp8(a, 1);
}

// CHECK-LABEL: @test_mov_dpp8_long(
// CHECK:      {{.*}}call{{.*}} i64 @llvm.amdgcn.mov.dpp8.i64(i64 %a, i32 1)
// CHECK-NEXT: store i64 %[[#]],
void test_mov_dpp8_long(global long* out, long a) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *out = __builtin_amdgcn_mov_dpp8(a, 1);
}

// CHECK-LABEL: @test_mov_dpp8_float(
// CHECK:      %[[BC:[0-9]+]] = bitcast float %a to i32
// CHECK-NEXT: %[[DPP_RET:[0-9]+]] = tail call{{.*}} i32 @llvm.amdgcn.mov.dpp8.i32(i32 %[[BC]], i32 1)
// CHECK-NEXT: store i32 %[[DPP_RET]],
void test_mov_dpp8_float(global float* out, float a) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *out = __builtin_amdgcn_mov_dpp8(a, 1);
}

// CHECK-LABEL: @test_mov_dpp8_double
// CHECK:      %[[BC:[0-9]+]] = bitcast double %x to i64
// CHECK-NEXT: %[[DPP_RET:[0-9]+]] = tail call{{.*}} i64 @llvm.amdgcn.mov.dpp8.i64(i64 %[[BC]], i32 1)
// CHECK-NEXT: store i64 %[[DPP_RET]],
void test_mov_dpp8_double(double x, global double *p) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *p = __builtin_amdgcn_mov_dpp8(x, 1);
}

// CHECK-LABEL: @test_mov_dpp8_short
// CHECK:      %[[ZEXT:[0-9]+]] = zext i16 %x to i32
// CHECK-NEXT: %[[DPP_RET:[0-9]+]] = tail call{{.*}} i32 @llvm.amdgcn.mov.dpp8.i32(i32 %[[ZEXT]], i32 1)
// CHECK-NEXT: %[[TRUNC:[0-9]+]] = trunc i32 %[[DPP_RET]] to i16
// CHECK-NEXT: store i16 %[[TRUNC]],
void test_mov_dpp8_short(short x, global short *p) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *p = __builtin_amdgcn_mov_dpp8(x, 1);
}

// CHECK-LABEL: @test_mov_dpp8_char
// CHECK:      %[[ZEXT:[0-9]+]] = zext i8 %x to i32
// CHECK-NEXT: %[[DPP_RET:[0-9]+]] = tail call{{.*}} i32 @llvm.amdgcn.mov.dpp8.i32(i32 %[[ZEXT]], i32 1)
// CHECK-NEXT: %[[TRUNC:[0-9]+]] = trunc i32 %[[DPP_RET]] to i8
// CHECK-NEXT: store i8 %[[TRUNC]],
void test_mov_dpp8_char(char x, global char *p) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *p = __builtin_amdgcn_mov_dpp8(x, 1);
}

// CHECK-LABEL: @test_mov_dpp8_half
// CHECK:      %[[LD:[0-9]+]] = load i16,
// CHECK:      %[[ZEXT:[0-9]+]] = zext i16 %[[LD]] to i32
// CHECK-NEXT: %[[DPP_RET:[0-9]+]] = tail call{{.*}} i32 @llvm.amdgcn.mov.dpp8.i32(i32 %[[ZEXT]], i32 1)
// CHECK-NEXT: %[[TRUNC:[0-9]+]] = trunc i32 %[[DPP_RET]] to i16
// CHECK-NEXT: store i16 %[[TRUNC]],
void test_mov_dpp8_half(half *x, global half *p) {
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_mov_dpp8))
    *p = __builtin_amdgcn_mov_dpp8(*x, 1);
}

// CHECK-LABEL: @test_s_memtime
// CHECK: {{.*}}call{{.*}} i64 @llvm.amdgcn.s.memtime()
void test_s_memtime(global ulong* out)
{
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_s_memtime))
    *out = __builtin_amdgcn_s_memtime();
}

// CHECK-LABEL: @test_groupstaticsize
// CHECK: {{.*}}call{{.*}} i32 @llvm.amdgcn.groupstaticsize()
void test_groupstaticsize(global uint* out)
{
  *out = __builtin_amdgcn_groupstaticsize();
}

// CHECK-LABEL: @test_ballot_wave32(
// CHECK: {{.*}}call{{.*}} i32 @llvm.amdgcn.ballot.i32(i1 %{{.+}})
void test_ballot_wave32(global uint* out, int a, int b)
{
  if (__builtin_amdgcn_is_invocable(__builtin_amdgcn_ballot_w32))
    *out = __builtin_amdgcn_ballot_w32(a == b);
}