63ca93c7d1
This patch renames the `OpenMPIRBuilderConfig` flags to reduce confusion over their meaning. `IsTargetCodegen` becomes `IsGPU`, whereas `IsEmbedded` becomes `IsTargetDevice`. The `-fopenmp-is-device` compiler option is also renamed to `-fopenmp-is-target-device` and the `omp.is_device` MLIR attribute is renamed to `omp.is_target_device`. Getters and setters of all these renamed properties are also updated accordingly. Many unit tests have been updated to use the new names, but an alias for the `-fopenmp-is-device` option is created so that external programs do not stop working after the name change. `IsGPU` is set when the target triple is AMDGCN or NVIDIA PTX, and it is only valid if `IsTargetDevice` is specified as well. `IsTargetDevice` is set by the `-fopenmp-is-target-device` compiler frontend option, which is only added to the OpenMP device invocation for offloading-enabled programs. Differential Revision: https://reviews.llvm.org/D154591
64 lines
1.9 KiB
C++
64 lines
1.9 KiB
C++
// REQUIRES: nvptx-registered-target
|
|
// RUN: %clang_cc1 -internal-isystem %S/Inputs/include -fopenmp -triple powerpc64le-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm-bc %s -o %t-ppc-host.bc
|
|
// RUN: %clang_cc1 -internal-isystem %S/../../lib/Headers/openmp_wrappers -include __clang_openmp_device_functions.h -internal-isystem %S/Inputs/include -fopenmp -triple nvptx64-nvidia-cuda -aux-triple powerpc64le-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm %s -fopenmp-is-target-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - | FileCheck %s
|
|
// RUN: %clang_cc1 -internal-isystem %S/../../lib/Headers/openmp_wrappers -DCMATH -include __clang_openmp_device_functions.h -internal-isystem %S/Inputs/include -fopenmp -triple nvptx64-nvidia-cuda -aux-triple powerpc64le-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm %s -fopenmp-is-target-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - | FileCheck %s
|
|
|
|
#ifdef CMATH
|
|
#include <cmath>
|
|
#else
|
|
#include <math.h>
|
|
#endif
|
|
|
|
// 4 calls to sincos(f), all translated to __nv_sincos calls:
|
|
|
|
// CHECK-NOT: _Z.sincos
|
|
// CHECK: call void @__nv_sincos(double
|
|
// CHECK-NOT: _Z.sincos
|
|
// CHECK: call void @__nv_sincosf(float
|
|
// CHECK-NOT: _Z.sincos
|
|
// CHECK: call void @__nv_sincos(double
|
|
// CHECK-NOT: _Z.sincos
|
|
// CHECK: call void @__nv_sincosf(float
|
|
// CHECK-NOT: _Z.sincos
|
|
|
|
// single precision wrapper
|
|
inline void sincos(float x, float* __restrict__ sin, float* __restrict__ cos)
|
|
{
|
|
sincosf(x, sin, cos);
|
|
}
|
|
|
|
template<typename T>
|
|
void test_sincos(T x)
|
|
{
|
|
T res_sin, res_cos;
|
|
|
|
#pragma omp target map(from: res_sin, res_cos)
|
|
{
|
|
sincos(x, &res_sin, &res_cos);
|
|
}
|
|
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
|
|
#if !defined(C_ONLY)
|
|
test_sincos<double>(0.0);
|
|
test_sincos<float>(0.0);
|
|
#endif
|
|
|
|
#pragma omp target
|
|
{
|
|
double s, c;
|
|
sincos(0, &s, &c);
|
|
}
|
|
|
|
#pragma omp target
|
|
{
|
|
float s, c;
|
|
sincosf(0.f, &s, &c);
|
|
}
|
|
|
|
return 0;
|
|
}
|