From 44cddd8eedb741051068078c6fd781b1cc2cc1aa Mon Sep 17 00:00:00 2001
From: liruoxuan <980773780@qq.com>
Date: Thu, 15 Sep 2022 15:25:46 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0Cholesky=E7=AE=97=E5=AD=90?=
 =?UTF-8?q?=E5=AF=B9=E9=9D=9E=E5=AF=B9=E7=A7=B0=E6=AD=A3=E5=AE=9A=E8=BE=93?=
 =?UTF-8?q?=E5=85=A5=E7=9A=84=E6=A0=A1=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../plugin/device/cpu/kernel/eigen/cholesky_cpu_kernel.cc      | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/cholesky_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/cholesky_cpu_kernel.cc
index 065c471c2d3..d8665add5e0 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/cholesky_cpu_kernel.cc
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/cholesky_cpu_kernel.cc
@@ -112,6 +112,9 @@ bool CholeskyCpuKernelMod::LaunchKernel(const std::vector<AddressPtr> &inputs, c
         }
       }
     } else {
+      if (!input.isApprox(input.transpose()) || llt.info() == Eigen::NumericalIssue) {
+        MS_LOG_EXCEPTION << "Cholesky expects symmetric positive definite matrices as inputs.";
+      }
       if (!upper_) {
         output = llt.matrixL();
       } else {