p89165320
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modified test cnnl and bang time

This commit is contained in:
xgqdut2016 2024-03-05 09:00:49 +00:00
parent 9aaf313c6f
commit 6ace4d8ae2
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/kernels/bang/test_bang_softmax.cc
View File

@ -81,10 +81,10 @@ void test_compareSoftmaxFp32(
inputGpu->setData(generator);
vector<float> inputData = inputGpu->copyout<float>();
double bangst, bangela;
bangst = get_walltime();
double cnnlst, cnnlela;
cnnlst = get_walltime();
bangRuntime->run(bangGraphCnnl);
bangela = 1000 * (get_walltime() - bangst);
cnnlela = 1000 * (get_walltime() - cnnlst);
auto outputGpu = gpuOp->getOutput();
auto outputGpu2Cpu = outputGpu->clone(cpuRuntime);
// bangSoftmax--------------------------------
@ -99,10 +99,10 @@ void test_compareSoftmaxFp32(
// bangInputGpu->setData(generator);
bangInputGpu->copyin(inputData);
double cnnlst, cnnlela;
cnnlst = get_walltime();
double bangst, bangela;
bangst = get_walltime();
bangRuntime->run(bangGraphBang);
cnnlela = 1000 * (get_walltime() - cnnlst);
bangela = 1000 * (get_walltime() - bangst);
auto bangOutputGpu = bangGpuOp->getOutput();
auto bangOutputGpu2Cpu = bangOutputGpu->clone(cpuRuntime);
// Check