ab02372c23
Summary: This patch is an attempt to get a clean run of `check-openmp` running on an NVPTX machine. I simply took the lists of tests that failed on my `sm_89` machine and disabled them or fixed them. A lot of these tests are disabled on AMDGPU already, so it makes sense that NVPTX fails. The others are simply problems with NVPTX optimized debugging which will need to be fixed. I opened an issue on one of them.
90 lines
2.4 KiB
C++
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;
|
|
}
|