diff --git a/include/armadillo_bits/op_inv_gen_meat.hpp b/include/armadillo_bits/op_inv_gen_meat.hpp index 6904ca21..82d261eb 100644 --- a/include/armadillo_bits/op_inv_gen_meat.hpp +++ b/include/armadillo_bits/op_inv_gen_meat.hpp @@ -203,7 +203,12 @@ op_inv_gen_full::apply_tiny(Mat& X) eT* outm = out.memptr(); if(N == 0) { return true; } - else if(N == 1) { outm[0] = eT(1) / Xm[0]; } + else if(N == 1) + { + if(Xm[0] == eT(0)) { return false; } + + outm[0] = eT(1) / Xm[0]; + } else if(N == 2) { const eT a = Xm[pos<0,0>::n2];