From 463738ccbe1ac26d841244aad7886dc92e639fe4 Mon Sep 17 00:00:00 2001 From: Rasmus Munk Larsen Date: Wed, 27 Apr 2016 12:26:18 -0700 Subject: Use computeProductBlockingSizes to compute blocking for both ShardByCol and ShardByRow cases. --- unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorContractionBlocking.h') 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(kc_, mc_, nc_, num_threads); } else { - if (kc_ && mc_ && nc_) { - mc_ = (((m / num_threads) + 15) / 16) * 16; - } + computeProductBlockingSizes(kc_, nc_, mc_, num_threads); } } -- cgit v1.2.3