diff options
| author | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-04 16:06:51 +0300 |
|---|---|---|
| committer | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-04 16:06:51 +0300 |
| commit | 5a43e6acaf1ddf62d4d875252aff5a8e89efadd4 (patch) | |
| tree | 45c4b51c718ce784936e966ced0233fe35778f38 | |
| parent | 2d4569b0a9d19c3301a7841ec889e24442c8a786 (diff) | |
debug
| -rw-r--r-- | src/amun/gpu/dl4mt/decoder.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/amun/gpu/dl4mt/decoder.h b/src/amun/gpu/dl4mt/decoder.h index 41d96a57..65338771 100644 --- a/src/amun/gpu/dl4mt/decoder.h +++ b/src/amun/gpu/dl4mt/decoder.h @@ -203,14 +203,23 @@ class Decoder { //std::cerr << "2Temp2_=" << Temp2_.Debug() << std::endl; Broadcast(Tanh(_1 + _2), Temp1_, SCU_, Temp2_, dBatchMapping_, maxLength); + HANDLE_ERROR( cudaStreamSynchronize(mblas::CudaStreamHandler::GetStream())); + std::cerr << "GetAlignedSourceContext1" << std::endl; //std::cerr << "w_.V_=" << w_.V_->Debug(0) << std::endl; //std::cerr << "3Temp1_=" << Temp1_.Debug(0) << std::endl; Prod(A_, *w_.V_, Temp1_, false, true); + HANDLE_ERROR( cudaStreamSynchronize(mblas::CudaStreamHandler::GetStream())); + std::cerr << "GetAlignedSourceContext2" << std::endl; mblas::Softmax(A_, dBatchMapping_, sentenceLengths, batchSize); + HANDLE_ERROR( cudaStreamSynchronize(mblas::CudaStreamHandler::GetStream())); + std::cerr << "GetAlignedSourceContext3" << std::endl; + mblas::WeightedMean(AlignedSourceContext, A_, SourceContext, dBatchMapping_); + HANDLE_ERROR( cudaStreamSynchronize(mblas::CudaStreamHandler::GetStream())); + std::cerr << "GetAlignedSourceContext4" << std::endl; /* std::cerr << "AlignedSourceContext=" << AlignedSourceContext.Debug() << std::endl; |