diff --git a/clang/lib/Lex/LiteralSupport.cpp b/clang/lib/Lex/LiteralSupport.cpp --- a/clang/lib/Lex/LiteralSupport.cpp +++ b/clang/lib/Lex/LiteralSupport.cpp @@ -943,9 +943,13 @@ // CUDA host and device may have different _Float16 support, therefore // allows f16 literals to avoid false alarm. + // When we compile for OpenMP target offloading on NVPTX, f16 suffix + // should also be supported. // ToDo: more precise check for CUDA. - if ((Target.hasFloat16Type() || LangOpts.CUDA) && s + 2 < ThisTokEnd && - s[1] == '1' && s[2] == '6') { + // TODO: AMDGPU might also support it in the future. + if ((Target.hasFloat16Type() || LangOpts.CUDA || + (LangOpts.OpenMPIsDevice && Target.getTriple().isNVPTX())) && + s + 2 < ThisTokEnd && s[1] == '1' && s[2] == '6') { s += 2; // success, eat up 2 characters. isFloat16 = true; continue; diff --git a/clang/test/OpenMP/float16_sema.cpp b/clang/test/OpenMP/float16_sema.cpp new file mode 100644 --- /dev/null +++ b/clang/test/OpenMP/float16_sema.cpp @@ -0,0 +1,10 @@ +// RUN: %clang_cc1 -fsyntax-only -x c++ -triple x86_64-unknown-linux-gnu -fopenmp -fopenmp-targets=nvptx64 -verify %s +// expected-no-diagnostics + +int foo() { +#pragma omp target + { + __fp16 a = -1.0f16; + } + return 0; +}