use copyFrom

2 files changed, 4 insertions, 24 deletions

diff --git a/src/amun/gpu/decoder/encoder_decoder.cu b/src/amun/gpu/decoder/encoder_decoder.cu
index 1444e240..b3c8c75f 100644
--- a/src/amun/gpu/decoder/encoder_decoder.cu
+++ b/src/amun/gpu/decoder/encoder_decoder.cu
@@ -379,12 +379,7 @@ void EncoderDecoder::AssembleBeamState(const Histories& histories,
   EDState& edOutState = outState.get<EDState>();
 
   thread_local mblas::Vector<unsigned> indices;
-  indices.newSize(beamStateIds.size());
-
-  mblas::copy(beamStateIds.data(),
-              beamStateIds.size(),
-              indices.data(),
-              cudaMemcpyHostToDevice);
+  indices.copyFrom(beamStateIds);
 
   CellState& cellOutStates = edOutState.GetStates();
   const CellState& cellInstates = edInState.GetStates();
diff --git a/src/amun/gpu/dl4mt/decoder.h b/src/amun/gpu/dl4mt/decoder.h
index 019de992..40dbe428 100644
--- a/src/amun/gpu/dl4mt/decoder.h
+++ b/src/amun/gpu/dl4mt/decoder.h
@@ -34,12 +34,7 @@ private:
               id = 1;
             }
           }
-          indices_.newSize(tids.size());
-
-          mblas::copy(tids.data(),
-              tids.size(),
-              indices_.data(),
-              cudaMemcpyHostToDevice);
+          indices_.copyFrom(tids);
 
           Assemble(Rows, *w_.E_, indices_);
         }
@@ -61,12 +56,7 @@ private:
               id = 1;
             }
           }
-          indices_.newSize(tids.size());
-
-          mblas::copy(tids.data(),
-              tids.size(),
-              indices_.data(),
-              cudaMemcpyHostToDevice);
+          indices_.copyFrom(tids);
 
           AssembleTopup(Rows, *w_.E_, indices_, numHypos, d_oldHypoIds);
 
@@ -216,12 +206,7 @@ private:
           unsigned batchSize = SourceContext.dim(3);
 
           std::vector<unsigned> hypo2Batch = histories.Hypo2Batch();
-          dHypo2Batch_.newSize(hypo2Batch.size());
-
-          mblas::copy(hypo2Batch.data(),
-              hypo2Batch.size(),
-              dHypo2Batch_.data(),
-              cudaMemcpyHostToDevice);
+          dHypo2Batch_.copyFrom(hypo2Batch);
 
           TIME_CMD("Prod3", Prod(/*h_[1],*/ Temp2_, *(HiddenState.output), *w_.W_));