#include <string>
#include "tensor_benchmarks.h"
Go to the source code of this file.
|
| BM_ContractionCPU (1, 2000, 500) |
|
| BM_ContractionCPU (1, 80, 13522) |
|
| BM_ContractionCPU (1500, 3, 512) |
|
| BM_ContractionCPU (2000, 1, 500) |
|
| BM_ContractionCPU (250, 3, 512) |
|
| BM_ContractionCPU (3200, 512, 4) |
|
| BM_ContractionCPU (3200, 512, 80) |
|
| BM_ContractionCPU (3200, 80, 512) |
|
| BM_ContractionCPU (512, 80, 13522) |
|
| BM_ContractionCPU (512, 80, 800) |
|
| BM_ContractionCPU (512, 800, 4) |
|
◆ BM_ContractionCPU
#define BM_ContractionCPU |
( |
|
D1, |
|
|
|
D2, |
|
|
|
D3 |
|
) |
| |
Value: static void BM_##Contraction##
_##D1##
x##D2##
x##D3(
int iters,
int Threads) { \
StopBenchmarkTiming(); \
CREATE_THREAD_POOL(Threads); \
BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
suite.contraction(iters); \
} \
BENCHMARK_RANGE(BM_##Contraction##
_##D1##
x##D2##
x##D3, 1, 32);
Definition at line 14 of file contraction_benchmarks_cpu.cc.
◆ CREATE_THREAD_POOL
#define CREATE_THREAD_POOL |
( |
|
threads | ) |
|
◆ EIGEN_USE_THREADS
#define EIGEN_USE_THREADS |
◆ BM_ContractionCPU() [1/11]
BM_ContractionCPU |
( |
1 |
, |
|
|
2000 |
, |
|
|
500 |
|
|
) |
| |
◆ BM_ContractionCPU() [2/11]
BM_ContractionCPU |
( |
1 |
, |
|
|
80 |
, |
|
|
13522 |
|
|
) |
| |
◆ BM_ContractionCPU() [3/11]
BM_ContractionCPU |
( |
1500 |
, |
|
|
3 |
, |
|
|
512 |
|
|
) |
| |
◆ BM_ContractionCPU() [4/11]
BM_ContractionCPU |
( |
2000 |
, |
|
|
1 |
, |
|
|
500 |
|
|
) |
| |
◆ BM_ContractionCPU() [5/11]
BM_ContractionCPU |
( |
250 |
, |
|
|
3 |
, |
|
|
512 |
|
|
) |
| |
◆ BM_ContractionCPU() [6/11]
BM_ContractionCPU |
( |
3200 |
, |
|
|
512 |
, |
|
|
4 |
|
|
) |
| |
◆ BM_ContractionCPU() [7/11]
BM_ContractionCPU |
( |
3200 |
, |
|
|
512 |
, |
|
|
80 |
|
|
) |
| |
◆ BM_ContractionCPU() [8/11]
BM_ContractionCPU |
( |
3200 |
, |
|
|
80 |
, |
|
|
512 |
|
|
) |
| |
◆ BM_ContractionCPU() [9/11]
BM_ContractionCPU |
( |
512 |
, |
|
|
80 |
, |
|
|
13522 |
|
|
) |
| |
◆ BM_ContractionCPU() [10/11]
BM_ContractionCPU |
( |
512 |
, |
|
|
80 |
, |
|
|
800 |
|
|
) |
| |
◆ BM_ContractionCPU() [11/11]
BM_ContractionCPU |
( |
512 |
, |
|
|
800 |
, |
|
|
4 |
|
|
) |
| |