8c1449a84d
This patch changes the kernels generated by OpenMP to have protected visibility. This is unlikely to change anything functionally. However, protected visibility better matches the behaviour of these GPU kernels. We do not expect any pending shared library load to preempt these kernels so we can specify a more restrictive visibility. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D136198
54 lines
1.7 KiB
C++
54 lines
1.7 KiB
C++
// REQUIRES: amdgpu-registered-target
|
|
|
|
// RUN: %clang_cc1 -fopenmp -x c++ -w -std=c++11 -triple x86_64-unknown-unknown -fopenmp-targets=amdgcn-amd-amdhsa -emit-llvm-bc %s -o %t-ppc-host.bc
|
|
// RUN: %clang_cc1 -fopenmp -x c++ -w -std=c++11 -triple amdgcn-amd-amdhsa -fopenmp-targets=amdgcn-amd-amdhsa -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -target-cpu gfx906 -o - | FileCheck %s
|
|
// expected-no-diagnostics
|
|
|
|
#ifndef HEADER
|
|
#define HEADER
|
|
|
|
int amdgcn_device_isa_selected() {
|
|
int threadCount = 0;
|
|
|
|
#pragma omp target map(tofrom \
|
|
: threadCount)
|
|
{
|
|
#pragma omp metadirective \
|
|
when(device = {isa("flat-address-space")} \
|
|
: parallel) default(single)
|
|
threadCount++;
|
|
}
|
|
|
|
return threadCount;
|
|
}
|
|
|
|
// CHECK: define weak_odr protected amdgpu_kernel void @__omp_offloading_{{.*}}amdgcn_device_isa_selected
|
|
// CHECK: user_code.entry:
|
|
// CHECK: call void @__kmpc_parallel_51
|
|
// CHECK-NOT: call i32 @__kmpc_single
|
|
// CHECK: ret void
|
|
|
|
int amdgcn_device_isa_not_selected() {
|
|
int threadCount = 0;
|
|
|
|
#pragma omp target map(tofrom \
|
|
: threadCount)
|
|
{
|
|
#pragma omp metadirective \
|
|
when(device = {isa("sse")} \
|
|
: parallel) \
|
|
when(device = {isa("another-unsupported-gpu-feature")} \
|
|
: parallel) default(single)
|
|
threadCount++;
|
|
}
|
|
|
|
return threadCount;
|
|
}
|
|
// CHECK: define weak_odr protected amdgpu_kernel void @__omp_offloading_{{.*}}amdgcn_device_isa_not_selected
|
|
// CHECK: user_code.entry:
|
|
// CHECK: call i32 @__kmpc_single
|
|
// CHECK-NOT: call void @__kmpc_parallel_51
|
|
// CHECK: ret void
|
|
|
|
#endif
|