Merge pull request #73 from kyungminlee/getri-fix

Fix inv to work with complex types
xtensor-stack · Jul 9, 2018 · d40eefc · d40eefc
2 parents f4100bd + 6b223cb
commit d40eefc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/xtensor-blas/xlapack.hpp b/include/xtensor-blas/xlapack.hpp
@@ -454,7 +454,7 @@ namespace lapack
             throw std::runtime_error("Could not find workspace size for getri.");
         }
 
-        work.resize(std::size_t(work[0]));
+        work.resize(std::size_t(std::real(work[0])));
 
         info = cxxlapack::getri<blas_index_t>(
             static_cast<blas_index_t>(A.shape()[0]),