mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-11 06:15:56 +03:00
temp()
This commit is contained in:
parent
7c63606b75
commit
3b0ea609e4
@ -39,12 +39,25 @@ std::string Expr::Debug() const
|
||||
}
|
||||
|
||||
///////////////////////////////////////////////////////
|
||||
#include "tensor_operators.h"
|
||||
|
||||
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();
|
||||
gSoftMax<<<2, 4>>>(data, 2, 4);
|
||||
|
||||
}
|
||||
|
||||
ExpressionGraph::ExpressionGraph(int cudaDevice)
|
||||
: stack_(new ChainableStack)
|
||||
{
|
||||
std::srand (time(NULL));
|
||||
cudaSetDevice(0);
|
||||
|
||||
temp();
|
||||
}
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user