diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-15 01:11:11 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-15 01:11:11 +0300 |
commit | 3ccde397f9f5550fdd7134a09bfa1b387d475b28 (patch) | |
tree | 48f4658b6564fb92ccb34a78be45db6f0fc97b45 /src/amun/gpu/mblas | |
parent | 71cfd04c863b6adeeb3877532c2dd4ad4dfaff61 (diff) |
use mblas::copy
Diffstat (limited to 'src/amun/gpu/mblas')
-rw-r--r-- | src/amun/gpu/mblas/nth_element.cu | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/amun/gpu/mblas/nth_element.cu b/src/amun/gpu/mblas/nth_element.cu index c4a426a1..31261f5c 100644 --- a/src/amun/gpu/mblas/nth_element.cu +++ b/src/amun/gpu/mblas/nth_element.cu @@ -156,9 +156,8 @@ void NthElement::getValueByKey(std::vector<float>& out, const mblas::Matrix &d_i // need a model with multiple scorers to test this method out.resize(d_breakdown.size()); - mblas::VectorWrapper<float> breakdownWrap(d_breakdown); - const mblas::MatrixWrapper<float> inWrap(d_in); - + //mblas::VectorWrapper<float> breakdownWrap(d_breakdown); + //const mblas::MatrixWrapper<float> inWrap(d_in); //gGetValueByKey<<<1, lastN_, 0, stream_>>> // (breakdownWrap, inWrap, h_res_idx, lastN_); /* @@ -168,8 +167,7 @@ void NthElement::getValueByKey(std::vector<float>& out, const mblas::Matrix &d_i << h_res.size() << endl; */ - HANDLE_ERROR( cudaMemcpyAsync(out.data(), d_breakdown.data(), d_breakdown.size() * sizeof(float), - cudaMemcpyDeviceToHost, mblas::CudaStreamHandler::GetStream()) ); + mblas::copy(d_breakdown.data(), d_breakdown.size(), out.data(), cudaMemcpyDeviceToHost); HANDLE_ERROR( cudaStreamSynchronize(mblas::CudaStreamHandler::GetStream())); } |