diff options
author | Rasmus Munk Larsen <rmlarsen@google.com> | 2016-04-27 12:26:18 -0700 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2016-04-27 12:26:18 -0700 |
commit | 463738ccbe1ac26d841244aad7886dc92e639fe4 (patch) | |
tree | da96494398a82526b4d4115f70ad565853623907 /unsupported/Eigen | |
parent | 3dddd341334975d4f71f4f7ca15c48aaa920c57d (diff) |
Use computeProductBlockingSizes to compute blocking for both ShardByCol and ShardByRow cases.
Diffstat (limited to 'unsupported/Eigen')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h index 3d3f6904f..5cf7b4f71 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h @@ -35,9 +35,7 @@ class TensorContractionBlocking { computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads); } else { - if (kc_ && mc_ && nc_) { - mc_ = (((m / num_threads) + 15) / 16) * 16; - } + computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, nc_, mc_, num_threads); } } |