clang/test/SemaCUDA/builtins.cu

// Tests that host and target builtins can be used in the same TU,
// have appropriate host/device attributes and that CUDA call
// restrictions are enforced. Also verify that non-target builtins can
// be used from both host and device functions.
//
// REQUIRES: x86-registered-target
// REQUIRES: nvptx-registered-target
// RUN: %clang_cc1 -triple x86_64-unknown-unknown \
// RUN:     -aux-triple nvptx64-unknown-cuda \
// RUN:     -fsyntax-only -verify=host %s
// RUN: %clang_cc1 -triple nvptx64-unknown-cuda -fcuda-is-device \
// RUN:     -aux-triple x86_64-unknown-unknown \
// RUN:     -target-cpu sm_80 -target-feature +ptx70 \
// RUN:     -fsyntax-only -verify=dev %s

#if !(defined(__amd64__) && defined(__PTX__))
#error "Expected to see preprocessor macros from both sides of compilation."
#endif

void hf() {
  int x = __builtin_ia32_rdtsc();
  int y = __nvvm_read_ptx_sreg_tid_x();
  // host-error@-1 {{reference to __device__ function '__nvvm_read_ptx_sreg_tid_x' in __host__ function}}
  x = __builtin_abs(1);
}

__attribute__((device)) void df() {
  int x = __nvvm_read_ptx_sreg_tid_x();
  int y = __builtin_ia32_rdtsc(); // dev-error {{reference to __host__ function '__builtin_ia32_rdtsc' in __device__ function}}
  x = __builtin_abs(1);
}

#if __CUDA_ARCH__ >= 800
__attribute__((device)) void nvvm_async_copy(__attribute__((address_space(3))) void* dst,
                                             __attribute__((address_space(1))) const void* src) {
  __nvvm_cp_async_ca_shared_global_4(dst, src);
  __nvvm_cp_async_ca_shared_global_8(dst, src);
  __nvvm_cp_async_ca_shared_global_16(dst, src);
  __nvvm_cp_async_cg_shared_global_16(dst, src);

  __nvvm_cp_async_ca_shared_global_4(dst, src, 2);
  __nvvm_cp_async_ca_shared_global_8(dst, src, 2);
  __nvvm_cp_async_ca_shared_global_16(dst, src, 2);
  __nvvm_cp_async_cg_shared_global_16(dst, src, 2);

  __nvvm_cp_async_ca_shared_global_4(dst, src, 2, 3); // dev-error {{too many arguments to function call}}
  __nvvm_cp_async_ca_shared_global_8(dst, src, 2, 3); // dev-error {{too many arguments to function call}}
  __nvvm_cp_async_ca_shared_global_16(dst, src, 2, 3); // dev-error {{too many arguments to function call}}
  __nvvm_cp_async_cg_shared_global_16(dst, src, 2, 3); // dev-error {{too many arguments to function call}}
}
#endif