@@ -94,9 +94,8 @@ TEST(JitKernel, vrelu) {
94
94
ker->Compute (x_data, ztgt_data, d);
95
95
}
96
96
auto ttgte = GetCurrentUS ();
97
- VLOG (3 ) << " Vec size " << d
98
- << " : refer takes: " << (trefe - trefs) / repeat
99
- << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
97
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
98
+ << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
100
99
for (int i = 0 ; i < d; ++i) {
101
100
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
102
101
}
@@ -127,9 +126,8 @@ TEST(JitKernel, vaddbias) {
127
126
}
128
127
auto ttgte = GetCurrentUS ();
129
128
130
- VLOG (3 ) << " Vec size " << d
131
- << " : refer takes: " << (trefe - trefs) / repeat
132
- << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
129
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
130
+ << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
133
131
for (int i = 0 ; i < d; ++i) {
134
132
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
135
133
}
@@ -235,11 +233,9 @@ TEST(JitKernel, vsigmoid) {
235
233
}
236
234
auto ttgte = GetCurrentUS ();
237
235
238
-
239
- VLOG (3 ) << " Vec size " << d
240
- << " : refer takes: " << (trefe - trefs) / repeat
241
- << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
242
- << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
236
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
237
+ << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
238
+ << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
243
239
for (int i = 0 ; i < d; ++i) {
244
240
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
245
241
}
@@ -297,11 +293,9 @@ TEST(JitKernel, vtanh) {
297
293
}
298
294
auto ttgte = GetCurrentUS ();
299
295
300
-
301
- VLOG (3 ) << " Vec size " << d
302
- << " : refer takes: " << (trefe - trefs) / repeat
303
- << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
304
- << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
296
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
297
+ << " us, better(jit exp) takes: " << (tmkle - tmkls) / repeat
298
+ << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
305
299
for (int i = 0 ; i < d; ++i) {
306
300
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
307
301
}
@@ -399,10 +393,9 @@ TEST(JitKernel, lstm) {
399
393
ker->ComputeCtHt (&step, &attr);
400
394
}
401
395
auto ttgte = GetCurrentUS ();
402
- VLOG (3 ) << " Vec size " << d
403
- << " : refer takes: " << (trefe - trefs) / repeat
404
- << " us, better(jit) takes: " << (tmkle - tmkls) / repeat
405
- << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
396
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
397
+ << " us, better(jit) takes: " << (tmkle - tmkls) / repeat
398
+ << " us, tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
406
399
}
407
400
}
408
401
@@ -476,7 +469,7 @@ TEST(JitKernel, vscal) {
476
469
}
477
470
auto si3 = GetCurrentUS ();
478
471
VLOG (3 ) << " Vec size 8 intr takes: " << (si1 - si0) / repeat
479
- << " us, inplace: " << (si3 - si2) / repeat << " us" ;
472
+ << " us, inplace: " << (si3 - si2) / repeat << " us" ;
480
473
}
481
474
#endif
482
475
@@ -497,9 +490,8 @@ TEST(JitKernel, vscal) {
497
490
#else
498
491
<< " us, "
499
492
#endif
500
- << " tgt takes: " << (ttgte - ttgts) / repeat
501
- << " us, tgt inplace takes: " << (ttgte1 - ttgts1) / repeat
502
- << " us" ;
493
+ << " tgt takes: " << (ttgte - ttgts) / repeat
494
+ << " us, tgt inplace takes: " << (ttgte1 - ttgts1) / repeat << " us" ;
503
495
for (int i = 0 ; i < d; ++i) {
504
496
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
505
497
}
@@ -573,7 +565,7 @@ TEST(JitKernel, vmul) {
573
565
#else
574
566
<< " us, "
575
567
#endif
576
- << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
568
+ << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
577
569
for (int i = 0 ; i < d; ++i) {
578
570
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
579
571
}
@@ -647,7 +639,7 @@ TEST(JitKernel, vadd) {
647
639
#else
648
640
<< " us, "
649
641
#endif
650
- << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
642
+ << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
651
643
for (int i = 0 ; i < d; ++i) {
652
644
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
653
645
}
@@ -697,10 +689,9 @@ TEST(JitKernel, vaddrelu) {
697
689
ker->Compute (x_data, y_data, ztgt_data, d);
698
690
}
699
691
auto ttgte = GetCurrentUS ();
700
- VLOG (3 ) << " Vec size " << d
701
- << " : refer takes: " << (trefe - trefs) / repeat
702
- << " us, better takes: " << (tmkle - tmkls) / repeat << " us, "
703
- << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
692
+ VLOG (3 ) << " Vec size " << d << " : refer takes: " << (trefe - trefs) / repeat
693
+ << " us, better takes: " << (tmkle - tmkls) / repeat << " us, "
694
+ << " tgt takes: " << (ttgte - ttgts) / repeat << " us" ;
704
695
for (int i = 0 ; i < d; ++i) {
705
696
EXPECT_NEAR (ztgt_data[i], zref_data[i], 1e-3 );
706
697
}
0 commit comments