GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-10 23:09:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use computeProductBlockingSizes to compute blocking for both ShardByCol and ShardByRow cases.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2016-04-27 12:26:18 -07:00
parent 3dddd34133
commit 463738ccbe
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h
View File

@ -35,9 +35,7 @@ class TensorContractionBlocking {
computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads); computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads);
} }
else { else {
if (kc_ && mc_ && nc_) { computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, nc_, mc_, num_threads);
mc_ = (((m / num_threads) + 15) / 16) * 16;
}
} }
} }