From 7cd9914f0e9f4efe300fda758e0b9d9d25e18fd6 Mon Sep 17 00:00:00 2001 From: Antonin Portelli Date: Thu, 8 Nov 2018 17:40:57 +0000 Subject: [PATCH] Hadrons: automatically resize output in MKL A2A matrix kernels --- Hadrons/A2AMatrix.hpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/Hadrons/A2AMatrix.hpp b/Hadrons/A2AMatrix.hpp index a2c2ffcf..2f958685 100644 --- a/Hadrons/A2AMatrix.hpp +++ b/Hadrons/A2AMatrix.hpp @@ -213,6 +213,10 @@ public: { static const ComplexD one(1., 0.), zero(0., 0.); + if ((res.rows() != a.rows()) or (res.cols() != b.cols())) + { + res.resize(a.rows(), b.cols()); + } if (Mat::Options == Eigen::RowMajor) { cblas_zgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a.rows(), b.cols(), @@ -234,6 +238,10 @@ public: { static const ComplexF one(1., 0.), zero(0., 0.); + if ((res.rows() != a.rows()) or (res.cols() != b.cols())) + { + res.resize(a.rows(), b.cols()); + } if (Mat::Options == Eigen::RowMajor) { cblas_cgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a.rows(), b.cols(),