Skip to content

Commit

Permalink
Print the rank to see when compute_error finishes
Browse files Browse the repository at this point in the history
  • Loading branch information
kmp5VT committed Jan 22, 2024
1 parent b6de171 commit d21ee08
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion unittest/ztensor_cp_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -105,12 +105,13 @@ TEST_CASE("ZCP") {
double diff = A1.compute_error(conv, 1e-5, 1, 100, true, 57);
std::cout << diff << std::endl;
std::cout << epsilon << std::endl;
std::cout << A1.get_factor_matrices()[1].extent(1) << std::endl;
CHECK(std::abs(diff) <= epsilon);
}
SECTION("ALS MODE = 4, Finite rank") {
CP_ALS<ztensor, zconv_class> A1(Z4);
conv.set_norm(norm4.real());
double diff = A1.compute_rank(70, conv, 1, true, 70);
double diff = A1.compute_rank(80, conv, 1, true, 80);
CHECK(std::abs(diff) <= epsilon);
}
#if BTAS_ENABLE_TUCKER_CP_UT
Expand Down

0 comments on commit d21ee08

Please sign in to comment.