Merge branch 'master' into complex-inshpect

eth-cscs · Dec 18, 2024 · 7f59c08 · 7f59c08
2 parents acf7afe + 0bd434c
commit 7f59c08
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/lapack/gpu/larft.cu b/src/lapack/gpu/larft.cu
@@ -9,6 +9,7 @@
 //
 
 #include <complex>
+#include <type_traits>
 
 #include <whip.hpp>
 
@@ -48,8 +49,20 @@ __global__ void fix_tau(const unsigned k, const T* tau, unsigned inctau, T* t, u
     t_ij = {};
   else if (i == j)
     t_ij = tau_j;
-  else
-    t_ij = -tau_j * t_ij;
+  else {
+#if defined(DLAF_WITH_HIP)
+    if constexpr (std::is_same_v<T, hipFloatComplex>) {
+      t_ij = hipCmulf(-tau_j, t_ij);
+    }
+    else if constexpr (std::is_same_v<T, hipDoubleComplex>) {
+      t_ij = hipCmul(-tau_j, t_ij);
+    }
+    else
+#endif
+    {
+      t_ij = -tau_j * t_ij;
+    }
+  }
 }
 }