GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-13 20:26:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Undo the block size change.

Browse Source 
.z *is* used by the EigenContractionKernelInternal().
This commit is contained in:
Artem Belevich 2019-12-09 11:10:29 -08:00
parent dbb703d44e
commit 8056a05b54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h
View File

@ -1335,7 +1335,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
const Index m_blocks = (m + 63) / 64; const Index m_blocks = (m + 63) / 64;
const Index n_blocks = (n + 63) / 64; const Index n_blocks = (n + 63) / 64;
const dim3 num_blocks(m_blocks, n_blocks, 1); const dim3 num_blocks(m_blocks, n_blocks, 1);
const dim3 block_size(8, 8, 1); const dim3 block_size(8, 8, 8);
LAUNCH_GPU_KERNEL((EigenContractionKernel<Scalar, Index, LhsMapper, RhsMapper, OutputMapper>), num_blocks, block_size, 0, device, lhs, rhs, output, m, n, k); LAUNCH_GPU_KERNEL((EigenContractionKernel<Scalar, Index, LhsMapper, RhsMapper, OutputMapper>), num_blocks, block_size, 0, device, lhs, rhs, output, m, n, k);
} }
}; };