75e648031c
Summary: The recent patch added `-nogpulib` to make these tests only pick up what was intentionally put into them. This had the effect of removing the dependency on the ROCm device libs which are needed for math. This test disables the complex math test, which is the only one that needed it, for the time being. In the future we will implement these and provide it via the GPU `libm` and pass it in the same way as the GPU `libc`.
89 lines
2.4 KiB
C++
89 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
|
|
|
|
#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;
|
|
}
|