contraction_benchmarks_cpu.cc File Reference

#include <string>
#include "tensor_benchmarks.h"

Macros
#define	EIGEN_USE_THREADS
#define	CREATE_THREAD_POOL(threads)
#define	BM_ContractionCPU(D1, D2, D3)

Functions
	BM_ContractionCPU (1, 2000, 500)
	BM_ContractionCPU (2000, 1, 500)
	BM_ContractionCPU (250, 3, 512)
	BM_ContractionCPU (1500, 3, 512)
	BM_ContractionCPU (512, 800, 4)
	BM_ContractionCPU (512, 80, 800)
	BM_ContractionCPU (512, 80, 13522)
	BM_ContractionCPU (1, 80, 13522)
	BM_ContractionCPU (3200, 512, 4)
	BM_ContractionCPU (3200, 512, 80)
	BM_ContractionCPU (3200, 80, 512)

Macro Definition Documentation

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);

CREATE_THREAD_POOL

#define CREATE_THREAD_POOL

(

threads

)

Value:

  Eigen::ThreadPool pool(threads);  \
  Eigen::ThreadPoolDevice device(&pool, threads);

EIGEN_USE_THREADS

#define EIGEN_USE_THREADS

Function Documentation

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
	)