devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve product test coverage at critical code-path boundaries

Browse Source 
libeigen/eigen!2285

Co-authored-by: Rasmus Munk Larsen <rmlarsen@gmail.com>
This commit is contained in:
Rasmus Munk Larsen
2026-03-12 12:32:06 -07:00
parent 3a2ba7c434
commit 5e478d3285
4 changed files with 287 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
test/product_threaded.cpp
View File

@@ -24,6 +24,54 @@ void test_parallelize_gemm() {
c_threaded.noalias() = a * b;
VERIFY_IS_APPROX(c, c_threaded);
Eigen::setGemmThreadPool(nullptr);
}
EIGEN_DECLARE_TEST(product_threaded) { CALL_SUBTEST(test_parallelize_gemm()); }
void test_parallelize_gemm_varied() {
constexpr int num_threads = 4;
ThreadPool pool(num_threads);
// Non-square float
{
MatrixXf a = MatrixXf::Random(512, 2048);
MatrixXf b = MatrixXf::Random(2048, 256);
MatrixXf c_serial(512, 256);
c_serial.noalias() = a * b;
Eigen::setGemmThreadPool(&pool);
MatrixXf c_threaded(512, 256);
c_threaded.noalias() = a * b;
Eigen::setGemmThreadPool(nullptr);
VERIFY_IS_APPROX(c_serial, c_threaded);
}
// Double
{
MatrixXd a = MatrixXd::Random(512, 512);
MatrixXd b = MatrixXd::Random(512, 512);
MatrixXd c_serial(512, 512);
c_serial.noalias() = a * b;
Eigen::setGemmThreadPool(&pool);
MatrixXd c_threaded(512, 512);
c_threaded.noalias() = a * b;
Eigen::setGemmThreadPool(nullptr);
VERIFY_IS_APPROX(c_serial, c_threaded);
}
// Complex double
{
MatrixXcd a = MatrixXcd::Random(256, 256);
MatrixXcd b = MatrixXcd::Random(256, 256);
MatrixXcd c_serial(256, 256);
c_serial.noalias() = a * b;
Eigen::setGemmThreadPool(&pool);
MatrixXcd c_threaded(256, 256);
c_threaded.noalias() = a * b;
Eigen::setGemmThreadPool(nullptr);
VERIFY_IS_APPROX(c_serial, c_threaded);
}
}
EIGEN_DECLARE_TEST(product_threaded) {
CALL_SUBTEST_1(test_parallelize_gemm());
CALL_SUBTEST_2(test_parallelize_gemm_varied());
}