From 60f25f70c8621da4e8694789f6fb6301e07d54a6 Mon Sep 17 00:00:00 2001 From: Artem Belevich Date: Wed, 4 Jan 2017 18:39:29 +0000 Subject: [CUDA] Pre-include sm_60 and sm_61 headers. CUDA-8.0 comes with new headers which nvcc pre-includes via cuda_runtime.h Clang now makes them available as well. Differential Revision: https://reviews.llvm.org/D28301 llvm-svn: 290982 --- clang/lib/Headers/__clang_cuda_runtime_wrapper.h | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'clang/lib/Headers/__clang_cuda_runtime_wrapper.h') diff --git a/clang/lib/Headers/__clang_cuda_runtime_wrapper.h b/clang/lib/Headers/__clang_cuda_runtime_wrapper.h index 205e15b40b5..931d44b6965 100644 --- a/clang/lib/Headers/__clang_cuda_runtime_wrapper.h +++ b/clang/lib/Headers/__clang_cuda_runtime_wrapper.h @@ -232,6 +232,11 @@ static inline __device__ void __brkpt(int __c) { __brkpt(); } // hardware, seems to generate faster machine code because ptxas can more easily // reason about our code. +#if CUDA_VERSION >= 8000 +#include "sm_60_atomic_functions.hpp" +#include "sm_61_intrinsics.hpp" +#endif + #undef __MATH_FUNCTIONS_HPP__ // math_functions.hpp defines ::signbit as a __host__ __device__ function. This -- cgit v1.2.3