GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable subtests which use device side malloc since this has been fixed in ROCm 5.2.

Browse Source
This commit is contained in:
Rohit Santhanam 2022-06-29 17:09:43 +00:00
parent 84cf3ff18d
commit 06a458a13d
1 changed files with 11 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/gpu_example.cu
View File

@ -70,7 +70,6 @@ struct MultiplyKernel {
template <typename T1, typename T2, typename T3>
void test_multiply(const T1& type1, const T2& type2, const T3& type3) {
const T1 A = T1::Random(type1.rows(), type1.cols());
const T2 B = T2::Random(type2.rows(), type2.cols());
T3 C;
@ -111,26 +110,20 @@ EIGEN_DECLARE_TEST(gpu_example)
// Call subtests with different sized/typed inputs.
CALL_SUBTEST( test_add(Eigen::Vector3f()) );
CALL_SUBTEST( test_add(Eigen::Matrix3d()) );
#if !defined(EIGEN_USE_HIP) // FIXME
CALL_SUBTEST( test_add(Eigen::MatrixX<int>(10, 10)) );
#endif
CALL_SUBTEST( test_add(Eigen::Array44f()) );
#if !defined(EIGEN_USE_HIP)
CALL_SUBTEST( test_add(Eigen::ArrayXd(20)) );
CALL_SUBTEST( test_add(Eigen::ArrayXXi(13, 17)) );
#endif
CALL_SUBTEST( test_multiply(Eigen::Matrix3d(),
Eigen::Matrix3d(),
Eigen::Matrix3d()) );
#if !defined(EIGEN_USE_HIP)
CALL_SUBTEST( test_multiply(Eigen::MatrixX<int>(10, 10),
Eigen::MatrixX<int>(10, 10),
Eigen::MatrixX<int>()) );
CALL_SUBTEST( test_multiply(Eigen::MatrixXf(12, 1),
Eigen::MatrixXf(1, 32),
Eigen::MatrixXf()) );
#endif
}
}