52 lines
2.3 KiB
Text
52 lines
2.3 KiB
Text
|
// 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
|