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 &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 {