Merge pull request #1318 from lattice/feature/multi-rhs

Feature/multi rhs
lattice · Oct 25, 2022 · 0a31b22 · 0a31b22
2 parents 3f3ed90 + a595b02
commit 0a31b22
Show file tree

Hide file tree

Showing 97 changed files with 3,482 additions and 3,864 deletions.
diff --git a/include/blas_helper.cuh b/include/blas_helper.cuh
@@ -5,6 +5,7 @@
 #include <convert.h>
 #include <float_vector.h>
 #include <array.h>
+#include <math_helper.cuh>
 
 //#define QUAD_SUM
 #ifdef QUAD_SUM
@@ -428,12 +429,12 @@ namespace quda
 
     template <template <typename...> class Functor,
               template <template <typename...> class, typename store_t, typename y_store_t, int, typename> class Blas,
-              bool mixed, typename T, typename x_store_t, typename V, typename... Args>
-    constexpr std::enable_if_t<mixed, void> instantiate(const T &a, const T &b, const T &c, V &x_, V &y_,
+              bool mixed, typename T, typename x_store_t, typename Vx, typename Vy, typename... Args>
+    constexpr std::enable_if_t<mixed, void> instantiate(const T &a, const T &b, const T &c, Vx &x_, Vy &y_,
                                                         Args &&... args)
     {
-      unwrap_t<V> &x(x_);
-      unwrap_t<V> &y(y_);
+      unwrap_t<Vx> &x(x_);
+      unwrap_t<Vy> &y(y_);
 
       if (y.Precision() < x.Precision()) errorQuda("Y precision %d not supported", y.Precision());