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

can call gSoftmax

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-16 14:50:14 +01:00
parent b6e4b8f3f3
commit 78444bc2eb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/expression_graph.cu
View File

@ -46,9 +46,9 @@ void temp()
std::vector<float> hVec({1,2, 4,3, 7,9, 7,3});
thrust::device_vector<float> dVec(8);
thrust::copy(hVec.begin(), hVec.end(), dVec.begin());
float *data = hVec.data();
float *data = thrust::raw_pointer_cast(dVec.data());
gSoftMax<<<2, 4>>>(data, 2, 4);
gSoftMax<<<4, 2, sizeof(float)>>>(data, 4, 2);
std::vector<float> hVec2(8);
thrust::copy(dVec.begin(), dVec.end(), hVec2.begin());
@ -66,6 +66,7 @@ ExpressionGraph::ExpressionGraph(int cudaDevice)
cudaSetDevice(0);
temp();
exit(0);
}
}