clang/test/CodeGenHIP/ballot.cpp

   1 // REQUIRES: amdgpu-registered-target
   2 // RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -aux-triple x86_64-pc-windows-msvc -target-cpu gfx900 -x hip -emit-llvm -fcuda-is-device -o - %s | FileCheck %s
   3 // RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -aux-triple x86_64-pc-windows-msvc -target-cpu gfx900 -x hip -S -fcuda-is-device -o - %s | FileCheck %s --check-prefix=GFX9
   4
   5 // Unlike OpenCL, HIP depends on the C++ interpration of "unsigned long", which
   6 // is 64 bits long on Linux and 32 bits long on Windows. The return type of the
   7 // ballot intrinsic needs to be a 64 bit integer on both platforms. This test
   8 // cross-compiles to Windows to confirm that the return type is indeed 64 bits
   9 // on Windows.
  10
  11 // CHECK-LABEL: @_Z3fooi
  12 // CHECK: call i64 @llvm.amdgcn.ballot.i64
  13
  14 // GFX9-LABEL: _Z3fooi:
  15 // GFX9: v_cmp_ne_u32_e64
  16
  17 #define __device__ __attribute__((device))
  18
  19 __device__ unsigned long long foo(int p) {
  20   return __builtin_amdgcn_ballot_w64(p);
  21 }