marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

temp()

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-16 15:08:09 +02:00
parent 3b0ea609e4
commit b6e4b8f3f3
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/expression_graph.cu
View File

@ -47,8 +47,16 @@ void temp()
thrust::device_vector<float> dVec(8); thrust::device_vector<float> dVec(8);
thrust::copy(hVec.begin(), hVec.end(), dVec.begin()); thrust::copy(hVec.begin(), hVec.end(), dVec.begin());
float *data = hVec.data(); float *data = hVec.data();
gSoftMax<<<2, 4>>>(data, 2, 4); gSoftMax<<<2, 4>>>(data, 2, 4);
std::vector<float> hVec2(8);
thrust::copy(dVec.begin(), dVec.end(), hVec2.begin());
cerr << "hVec2=";
for (size_t i = 0; i < hVec.size(); ++i) {
cerr << hVec2[i] << " ";
}
cerr << endl;
} }
ExpressionGraph::ExpressionGraph(int cudaDevice) ExpressionGraph::ExpressionGraph(int cudaDevice)