diff options
author | Yaxun Liu <Yaxun.Liu@amd.com> | 2018-04-25 13:33:19 +0000 |
---|---|---|
committer | Yaxun Liu <Yaxun.Liu@amd.com> | 2018-04-25 13:33:19 +0000 |
commit | 9061e4f48627f01316ddc5887f3dd23225fdd500 (patch) | |
tree | e47efbfb7a9fc19e0b57416eb8cf06eb38eac00d /clang/lib/Frontend/InitPreprocessor.cpp | |
parent | 0d97c225cd713474767b9bc7b6dd7219638073ad (diff) | |
download | llvm-9061e4f48627f01316ddc5887f3dd23225fdd500.zip llvm-9061e4f48627f01316ddc5887f3dd23225fdd500.tar.gz llvm-9061e4f48627f01316ddc5887f3dd23225fdd500.tar.bz2 |
[HIP] Add predefined macros __HIPCC__ and __HIP_DEVICE_COMPILE__
Differential Revision: https://reviews.llvm.org/D45441
llvm-svn: 330824
Diffstat (limited to 'clang/lib/Frontend/InitPreprocessor.cpp')
-rw-r--r-- | clang/lib/Frontend/InitPreprocessor.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/clang/lib/Frontend/InitPreprocessor.cpp b/clang/lib/Frontend/InitPreprocessor.cpp index ebb3f26..192862d 100644 --- a/clang/lib/Frontend/InitPreprocessor.cpp +++ b/clang/lib/Frontend/InitPreprocessor.cpp @@ -473,8 +473,12 @@ static void InitializeStandardPredefinedMacros(const TargetInfo &TI, Builder.defineMacro("__ASSEMBLER__"); if (LangOpts.CUDA && !LangOpts.HIP) Builder.defineMacro("__CUDA__"); - if (LangOpts.HIP) + if (LangOpts.HIP) { Builder.defineMacro("__HIP__"); + Builder.defineMacro("__HIPCC__"); + if (LangOpts.CUDAIsDevice) + Builder.defineMacro("__HIP_DEVICE_COMPILE__"); + } } /// Initialize the predefined C++ language feature test macros defined in @@ -1033,7 +1037,7 @@ static void InitializePredefinedMacros(const TargetInfo &TI, } // CUDA device path compilaton - if (LangOpts.CUDAIsDevice) { + if (LangOpts.CUDAIsDevice && !LangOpts.HIP) { // The CUDA_ARCH value is set for the GPU target specified in the NVPTX // backend's target defines. Builder.defineMacro("__CUDA_ARCH__"); |