maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Emit OpenCL metadata when targeting SPIR-V 

This is required for converting function calls such as get_global_id()
into SPIR-V builtins.

Differential Revision: https://reviews.llvm.org/D123049
This commit is contained in:
Shangwu Yao 2022-04-04 16:52:46 +00:00
parent 1acba8a4b5
commit 15a1769631
3 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clang/lib/CodeGen/CodeGenModule.cpp
View File

@ -784,7 +784,7 @@ void CodeGenModule::Release() {
LangOpts.OpenMP); LangOpts.OpenMP);
// Emit OpenCL specific module metadata: OpenCL/SPIR version. // Emit OpenCL specific module metadata: OpenCL/SPIR version.
if (LangOpts.OpenCL) { if (LangOpts.OpenCL || (LangOpts.CUDAIsDevice && getTriple().isSPIRV())) {
EmitOpenCLMetadata(); EmitOpenCLMetadata();
// Emit SPIR version. // Emit SPIR version.
if (getTriple().isSPIR()) { if (getTriple().isSPIR()) {

4
clang/lib/Frontend/CompilerInvocation.cpp
View File

@ -3328,6 +3328,10 @@ void CompilerInvocation::setLangDefaults(LangOptions &Opts, InputKind IK,
// whereas respecting contract flag in backend. // whereas respecting contract flag in backend.
Opts.setDefaultFPContractMode(LangOptions::FPM_FastHonorPragmas); Opts.setDefaultFPContractMode(LangOptions::FPM_FastHonorPragmas);
} else if (Opts.CUDA) { } else if (Opts.CUDA) {
if (T.isSPIRV()) {
// Emit OpenCL version metadata in LLVM IR when targeting SPIR-V.
Opts.OpenCLVersion = 200;
}
// Allow fuse across statements disregarding pragmas. // Allow fuse across statements disregarding pragmas.
Opts.setDefaultFPContractMode(LangOptions::FPM_Fast); Opts.setDefaultFPContractMode(LangOptions::FPM_Fast);
} }

3
clang/test/CodeGenCUDASPIRV/kernel-cc.cu
View File

@ -7,3 +7,6 @@
// CHECK: define spir_kernel void @_Z6kernelv() // CHECK: define spir_kernel void @_Z6kernelv()
__attribute__((global)) void kernel() { return; } __attribute__((global)) void kernel() { return; }
// CHECK: !opencl.ocl.version = !{[[OCL:![0-9]+]]}
// CHECK: [[OCL]] = !{i32 2, i32 0}