llvm-project/offload/test/offloading/std_complex_arithmetic.cpp
Johannes Doerfert 330d8983d2
[Offload] Move /openmp/libomptarget to /offload (#75125)
In a nutshell, this moves our libomptarget code to populate the offload
subproject.

With this commit, users need to enable the new LLVM/Offload subproject
as a runtime in their cmake configuration.
No further changes are expected for downstream code.

Tests and other components still depend on OpenMP and have also not been
renamed. The results below are for a build in which OpenMP and Offload
are enabled runtimes. In addition to the pure `git mv`, we needed to
adjust some CMake files. Nothing is intended to change semantics.

```
ninja check-offload
```
Works with the X86 and AMDGPU offload tests

```
ninja check-openmp
```
Still works but doesn't build offload tests anymore.

```
ls install/lib
```
Shows all expected libraries, incl.
- `libomptarget.devicertl.a`
- `libomptarget-nvptx-sm_90.bc`
- `libomptarget.rtl.amdgpu.so` -> `libomptarget.rtl.amdgpu.so.18git`
- `libomptarget.so` -> `libomptarget.so.18git`

Fixes: https://github.com/llvm/llvm-project/issues/75124

---------

Co-authored-by: Saiyedul Islam <Saiyedul.Islam@amd.com>
2024-04-22 09:51:33 -07:00

90 lines
2.4 KiB
C++

// RUN: %libomptarget-compilexx-generic && %libomptarget-run-generic
// RUN: %libomptarget-compilexx-generic -O3 && %libomptarget-run-generic
// RUN: %libomptarget-compilexx-generic -O3 -ffast-math && \
// RUN: %libomptarget-run-generic
// FIXME: This fails to link due to missing math symbols. We should provide the
// needed math functions in the GPU `libm` and require the GPU C library.
// UNSUPPORTED: amdgcn-amd-amdhsa
// UNSUPPORTED: nvptx64-nvidia-cuda-LTO
#include <cassert>
#include <complex>
#include <iostream>
template <typename T> void test_map() {
std::complex<T> a(0.2, 1), a_check;
#pragma omp target map(from : a_check)
{ a_check = a; }
assert(std::abs(a - a_check) < 1e-6);
}
template <typename RT, typename AT, typename BT> void test_plus(AT a, BT b) {
std::complex<RT> c, c_host;
c_host = a + b;
#pragma omp target map(from : c)
{ c = a + b; }
assert(std::abs(c - c_host) < 1e-6);
}
template <typename RT, typename AT, typename BT> void test_minus(AT a, BT b) {
std::complex<RT> c, c_host;
c_host = a - b;
#pragma omp target map(from : c)
{ c = a - b; }
assert(std::abs(c - c_host) < 1e-6);
}
template <typename RT, typename AT, typename BT> void test_mul(AT a, BT b) {
std::complex<RT> c, c_host;
c_host = a * b;
#pragma omp target map(from : c)
{ c = a * b; }
assert(std::abs(c - c_host) < 1e-6);
}
template <typename RT, typename AT, typename BT> void test_div(AT a, BT b) {
std::complex<RT> c, c_host;
c_host = a / b;
#pragma omp target map(from : c)
{ c = a / b; }
assert(std::abs(c - c_host) < 1e-6);
}
template <typename T> void test_complex() {
test_map<T>();
test_plus<T>(std::complex<T>(0, 1), std::complex<T>(0.5, 0.3));
test_plus<T>(std::complex<T>(0, 1), T(0.5));
test_plus<T>(T(0.5), std::complex<T>(0, 1));
test_minus<T>(std::complex<T>(0, 1), std::complex<T>(0.5, 0.3));
test_minus<T>(std::complex<T>(0, 1), T(0.5));
test_minus<T>(T(0.5), std::complex<T>(0, 1));
test_mul<T>(std::complex<T>(0, 1), std::complex<T>(0.5, 0.3));
test_mul<T>(std::complex<T>(0, 1), T(0.5));
test_mul<T>(T(0.5), std::complex<T>(0, 1));
test_div<T>(std::complex<T>(0, 1), std::complex<T>(0.5, 0.3));
test_div<T>(std::complex<T>(0, 1), T(0.5));
test_div<T>(T(0.5), std::complex<T>(0, 1));
}
int main() {
std::cout << "Testing float" << std::endl;
test_complex<float>();
std::cout << "Testing double" << std::endl;
test_complex<double>();
return 0;
}