devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Turned on the contraction benchmarks for fp16

Browse Source
This commit is contained in:
Benoit Steiner
2016-04-12 14:11:52 -07:00
parent e3a184785c
commit 53121c0119
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bench/tensors/tensor_benchmarks_fp16_gpu.cu
View File

@@ -48,11 +48,11 @@ BM_FuncGPU(colReduction);
BENCHMARK_RANGE(BM_##FUNC##_##D1##x##D2##x##D3, 10, 5000);
/*BM_FuncWithInputDimsGPU(contraction, N, N, N);
BM_FuncWithInputDimsGPU(contraction, N, N, N);
BM_FuncWithInputDimsGPU(contraction, 64, N, N);
BM_FuncWithInputDimsGPU(contraction, N, 64, N);
BM_FuncWithInputDimsGPU(contraction, N, N, 64);
*/
// Convolutions
#define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \