[Openmp-commits] [openmp] 0e68f48 - [OpenMP] add a offload test involving std::complex

[Ye Luo via Openmp-commits]

Author: Ye Luo
Date: 2022-09-03T13:28:11-05:00
New Revision: 0e68f483d48d5d468addfd3a8c68a8862aa60222

URL: https://github.com/llvm/llvm-project/commit/0e68f483d48d5d468addfd3a8c68a8862aa60222
DIFF: https://github.com/llvm/llvm-project/commit/0e68f483d48d5d468addfd3a8c68a8862aa60222.diff

LOG: [OpenMP] add a offload test involving std::complex

Taken from the https://github.com/llvm/llvm-project/issues/57064 reproducer.

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D133258

Added: 
    openmp/libomptarget/test/offloading/std_complex_arithmetic.cpp

Modified: 
    openmp/libomptarget/test/offloading/bug49021.cpp

Removed: 
    


################################################################################
diff  --git a/openmp/libomptarget/test/offloading/bug49021.cpp b/openmp/libomptarget/test/offloading/bug49021.cpp
index 57c9b9c7d9152..feeb48e73e6c4 100644
--- a/openmp/libomptarget/test/offloading/bug49021.cpp
+++ b/openmp/libomptarget/test/offloading/bug49021.cpp
@@ -68,7 +68,7 @@ template <typename T> int test_reduction() {
   return 0;
 }
 
-template <typename T> int test_complex() {
+template <typename T> int test_POD() {
   int ret = 0;
   ret |= test_map<T>();
   ret |= test_reduction<T>();
@@ -78,8 +78,8 @@ template <typename T> int test_complex() {
 int main() {
   int ret = 0;
   std::cout << "Testing float" << std::endl;
-  ret |= test_complex<float>();
+  ret |= test_POD<float>();
   std::cout << "Testing double" << std::endl;
-  ret |= test_complex<double>();
+  ret |= test_POD<double>();
   return ret;
 }

diff  --git a/openmp/libomptarget/test/offloading/std_complex_arithmetic.cpp b/openmp/libomptarget/test/offloading/std_complex_arithmetic.cpp
new file mode 100644
index 0000000000000..b06d605d29f90
--- /dev/null
+++ b/openmp/libomptarget/test/offloading/std_complex_arithmetic.cpp
@@ -0,0 +1,84 @@
+// 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
+
+#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;
+}