Skip to content

P
Paddle

吃玉米的猫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 0
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 9d7c3b18 编写于 作者: M minqiyang
浏览文件
操作

Polish code 

test=develop
上级 2b430ada
隐藏空白更改
内联 并排
Showing with 21 addition and 30 deletion
paddle/fluid/operators/math/jit_kernel_test.cc
浏览文件 @ 9d7c3b18
...@@ -94,9 +94,8 @@ TEST(JitKernel, vrelu) { ...@@ -94,9 +94,8 @@ TEST(JitKernel, vrelu) {
ker->Compute(x_data, ztgt_data, d); ker->Compute(x_data, ztgt_data, d);
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -127,9 +126,8 @@ TEST(JitKernel, vaddbias) { ...@@ -127,9 +126,8 @@ TEST(JitKernel, vaddbias) {
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -235,11 +233,9 @@ TEST(JitKernel, vsigmoid) { ...@@ -235,11 +233,9 @@ TEST(JitKernel, vsigmoid) {
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
VLOG(3) << "Vec size " << d << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
<< " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -297,11 +293,9 @@ TEST(JitKernel, vtanh) { ...@@ -297,11 +293,9 @@ TEST(JitKernel, vtanh) {
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
VLOG(3) << "Vec size " << d << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
<< " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -399,10 +393,9 @@ TEST(JitKernel, lstm) { ...@@ -399,10 +393,9 @@ TEST(JitKernel, lstm) {
ker->ComputeCtHt(&step, &attr); ker->ComputeCtHt(&step, &attr);
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, better(jit) takes: " << (tmkle - tmkls) / repeat
<< " us, better(jit) takes: " << (tmkle - tmkls) / repeat << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< " us, tgt takes: " << (ttgte - ttgts) / repeat << " us";
} }
} }
...@@ -476,7 +469,7 @@ TEST(JitKernel, vscal) { ...@@ -476,7 +469,7 @@ TEST(JitKernel, vscal) {
} }
auto si3 = GetCurrentUS(); auto si3 = GetCurrentUS();
VLOG(3) << "Vec size 8 intr takes: " << (si1 - si0) / repeat VLOG(3) << "Vec size 8 intr takes: " << (si1 - si0) / repeat
<< " us, inplace: " << (si3 - si2) / repeat << " us"; << " us, inplace: " << (si3 - si2) / repeat << " us";
} }
#endif #endif
...@@ -497,9 +490,8 @@ TEST(JitKernel, vscal) { ...@@ -497,9 +490,8 @@ TEST(JitKernel, vscal) {
#else #else
<< " us, " << " us, "
#endif #endif
<< "tgt takes: " << (ttgte - ttgts) / repeat << "tgt takes: " << (ttgte - ttgts) / repeat
<< "us, tgt inplace takes: " << (ttgte1 - ttgts1) / repeat << "us, tgt inplace takes: " << (ttgte1 - ttgts1) / repeat << " us";
<< " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -573,7 +565,7 @@ TEST(JitKernel, vmul) { ...@@ -573,7 +565,7 @@ TEST(JitKernel, vmul) {
#else #else
<< " us, " << " us, "
#endif #endif
<< "tgt takes: " << (ttgte - ttgts) / repeat << " us"; << "tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -647,7 +639,7 @@ TEST(JitKernel, vadd) { ...@@ -647,7 +639,7 @@ TEST(JitKernel, vadd) {
#else #else
<< " us, " << " us, "
#endif #endif
<< "tgt takes: " << (ttgte - ttgts) / repeat << " us"; << "tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
...@@ -697,10 +689,9 @@ TEST(JitKernel, vaddrelu) { ...@@ -697,10 +689,9 @@ TEST(JitKernel, vaddrelu) {
ker->Compute(x_data, y_data, ztgt_data, d); ker->Compute(x_data, y_data, ztgt_data, d);
} }
auto ttgte = GetCurrentUS(); auto ttgte = GetCurrentUS();
VLOG(3) << "Vec size " << d VLOG(3) << "Vec size " << d << ": refer takes: " << (trefe - trefs) / repeat
<< ": refer takes: " << (trefe - trefs) / repeat << " us, better takes: " << (tmkle - tmkls) / repeat << " us, "
<< " us, better takes: " << (tmkle - tmkls) / repeat << " us, " << "tgt takes: " << (ttgte - ttgts) / repeat << " us";
<< "tgt takes: " << (ttgte - ttgts) / repeat << " us";
for (int i = 0; i < d; ++i) { for (int i = 0; i < d; ++i) {
EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3); EXPECT_NEAR(ztgt_data[i], zref_data[i], 1e-3);
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或