marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-11 06:15:56 +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::copy(hVec.begin(), hVec.end(), dVec.begin());
float *data = hVec.data();
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)