phynexis-deps/armadillo-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more efficient handling of aliasing in default case

Browse Source
This commit is contained in:
conrad
2022-05-09 14:56:37 +10:00
parent 950540ef59
commit d5760f310c
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/armadillo_bits/glue_solve_meat.hpp
+13 -2
View File
@@ -117,7 +117,18 @@ glue_solve_gen_full::apply(Mat<eT>& actual_out, const Base<eT,T1>& A_expr, const
arma_debug_check( (fast && refine ), "solve(): options 'fast' and 'refine' are mutually exclusive" );
arma_debug_check( (no_sympd && likely_sympd), "solve(): options 'no_sympd' and 'likely_sympd' are mutually exclusive" );
Mat<eT> out;
bool is_alias = true;
if(is_Mat<T1>::value && is_Mat<T2>::value)
{
const quasi_unwrap<T1> UA( A_expr.get_ref() );
const quasi_unwrap<T2> UB( B_expr.get_ref() );
if( (UA.is_alias(actual_out) == false) && (UB.is_alias(actual_out) == false) ) { is_alias = false; }
}
Mat<eT> tmp;
Mat<eT>& out = (is_alias) ? tmp : actual_out;
Mat<eT> A = A_expr.get_ref();
@@ -329,7 +340,7 @@ glue_solve_gen_full::apply(Mat<eT>& actual_out, const Base<eT,T1>& A_expr, const
status = auxlib::solve_approx_svd(out, A, B_expr.get_ref()); // A is overwritten
}
actual_out.steal_mem(out);
if(is_alias) { actual_out.steal_mem(out); }
return status;
}