diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-20 19:31:57 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-20 19:31:57 +0300 |
commit | d88cbf7f823eee5b10df98ed98c3681ffd0a7c3b (patch) | |
tree | cb3048ac38a9c38be1f01661e1e9af52c2fe4f87 | |
parent | 04fe617903cb3c8fe856d48183659ceba9f9a3e7 (diff) |
shrink size of shape in Broadcast
-rw-r--r-- | src/amun/gpu/mblas/matrix_functions.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/amun/gpu/mblas/matrix_functions.h b/src/amun/gpu/mblas/matrix_functions.h index 43233629..02eee41d 100644 --- a/src/amun/gpu/mblas/matrix_functions.h +++ b/src/amun/gpu/mblas/matrix_functions.h @@ -192,7 +192,7 @@ Matrix& Broadcast(Functor functor, std::cerr << "srcSize=" << srcSize << " " << activeBatchMaxLength << std::endl; //std::cerr << std::endl; - Shape shape(srcSize, cols, sumOfBeamSizes, 1); + Shape shape(activeBatchMaxLength, cols, sumOfBeamSizes, 1); unsigned size = shape.size(); unsigned threads = std::min(MAX_THREADS, size); unsigned blocks = (size / threads) + ((size % threads == 0) ? 0 : 1); |