jiuyuan
/
InfiniTensor
mirror of https://github.com/InfiniTensor/InfiniTensor
6
0
Fork 21
Code Issues Projects Releases Wiki Activity

fix rope test

This commit is contained in:
kilinchange 2024-08-07 09:41:18 +08:00
parent b9699b0e7a
commit 068b7e51d6
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/kernels/cuda/test_cuda_rope.cc
View File

@ -27,10 +27,10 @@ TEST(RoPE, Cuda) {
auto oCpu = gCpu->cloneTensor(op->getOutputs()[0]);
EXPECT_TRUE(oCpu->equalData(vector<float>{
1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773,
1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773,
1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773,
1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773, 1.381773,
1.381773, 1.381773, 1.381773, 1.381773}));
0.540302, 0.647906, 0.731761, 0.796458, 0.846009, 0.883756, 0.912396,
0.934062, 0.950415, 0.962739, 0.972014, 0.978989, 0.98423, 0.988167,
0.991122, 0.99334, 0.995004, 0.996253, 0.99719, 0.997892, 0.998419,
0.998815, 0.999111, 0.999333, 0.9995, 0.999625, 0.999719, 0.999789,
0.999842, 0.999881, 0.999911, 0.999933}));
}
} // namespace infini