add Row()

1 files changed, 9 insertions, 0 deletions

diff --git a/src/amun/gpu/mblas/matrix_wrapper.h b/src/amun/gpu/mblas/matrix_wrapper.h
index aecb5729..4e50a9aa 100644
--- a/src/amun/gpu/mblas/matrix_wrapper.h
+++ b/src/amun/gpu/mblas/matrix_wrapper.h
@@ -1,5 +1,6 @@
 #pragma once
 #include "matrix.h"
+#include "vector_wrapper.h"
 
 namespace amunmt {
 namespace GPU {
@@ -268,6 +269,14 @@ public:
     return ind;
   }
 
+  __device__
+  VectorWrapper<T> Row(unsigned row)
+  {
+    T &ele = (*this)(row);
+    VectorWrapper<T> ret(&ele, dim(1));
+    return ret;
+  }
+
   std::string Debug() const
   {
     std::stringstream strm;