int main (int, const char **)
{
typedef float ScalarType; //feel free to change this to 'double' if supported by your hardware
typedef boost::numeric::ublas::matrix<ScalarType> MatrixType;
typedef viennacl::matrix<ScalarType, viennacl::row_major> VCLMatrixType;
std::size_t dim_large = 5;
std::size_t dim_small = 3;
//
// Setup ublas objects and fill with data:
//
MatrixType ublas_A(dim_large, dim_large);
MatrixType ublas_B(dim_small, dim_small);
MatrixType ublas_C(dim_large, dim_small);
MatrixType ublas_D(dim_small, dim_large);
for (std::size_t i=0; i<ublas_A.size1(); ++i)
for (std::size_t j=0; j<ublas_A.size2(); ++j)
ublas_A(i,j) = static_cast<ScalarType>((i+1) + (j+1)*(i+1));
for (std::size_t i=0; i<ublas_B.size1(); ++i)
for (std::size_t j=0; j<ublas_B.size2(); ++j)
ublas_B(i,j) = static_cast<ScalarType>((i+1) + (j+1)*(i+1));
for (std::size_t i=0; i<ublas_C.size1(); ++i)
for (std::size_t j=0; j<ublas_C.size2(); ++j)
ublas_C(i,j) = static_cast<ScalarType>((j+2) + (j+1)*(i+1));
for (std::size_t i=0; i<ublas_D.size1(); ++i)
for (std::size_t j=0; j<ublas_D.size2(); ++j)
ublas_D(i,j) = static_cast<ScalarType>((j+2) + (j+1)*(i+1));
//
// Extract submatrices using the ranges in ublas
//
boost::numeric::ublas::range ublas_r1(0, dim_small); //the first 'dim_small' entries
boost::numeric::ublas::range ublas_r2(dim_large - dim_small, dim_large); //the last 'dim_small' entries
boost::numeric::ublas::matrix_range<MatrixType> ublas_A_sub1(ublas_A, ublas_r1, ublas_r1); //upper left part of A
boost::numeric::ublas::matrix_range<MatrixType> ublas_A_sub2(ublas_A, ublas_r2, ublas_r2); //lower right part of A
boost::numeric::ublas::matrix_range<MatrixType> ublas_C_sub(ublas_C, ublas_r1, ublas_r1); //upper left part of C
boost::numeric::ublas::matrix_range<MatrixType> ublas_D_sub(ublas_D, ublas_r1, ublas_r1); //upper left part of D
//
// Setup ViennaCL objects
//
VCLMatrixType vcl_A(dim_large, dim_large);
VCLMatrixType vcl_B(dim_small, dim_small);
VCLMatrixType vcl_C(dim_large, dim_small);
VCLMatrixType vcl_D(dim_small, dim_large);
viennacl::copy(ublas_A, vcl_A);
viennacl::copy(ublas_B, vcl_B);
viennacl::copy(ublas_C, vcl_C);
viennacl::copy(ublas_D, vcl_D);
//
// Extract submatrices using the ranges in ViennaCL
//
viennacl::range vcl_r1(0, dim_small); //the first 'dim_small' entries
viennacl::range vcl_r2(dim_large - dim_small, dim_large); //the last 'dim_small' entries
viennacl::matrix_range<VCLMatrixType> vcl_A_sub1(vcl_A, vcl_r1, vcl_r1); //upper left part of A
viennacl::matrix_range<VCLMatrixType> vcl_A_sub2(vcl_A, vcl_r2, vcl_r2); //lower right part of A
viennacl::matrix_range<VCLMatrixType> vcl_C_sub(vcl_C, vcl_r1, vcl_r1); //upper left part of C
viennacl::matrix_range<VCLMatrixType> vcl_D_sub(vcl_D, vcl_r1, vcl_r1); //upper left part of D
//
// Copy from ublas to submatrices and back:
//
ublas_A_sub1 = ublas_B;
viennacl::copy(ublas_B, vcl_A_sub1);
viennacl::copy(vcl_A_sub1, ublas_B);
//
// Addition:
//
// range to range:
ublas_A_sub2 += ublas_A_sub2;
vcl_A_sub2 += vcl_A_sub2;
// range to matrix:
ublas_B += ublas_A_sub2;
vcl_B += vcl_A_sub2;
//
// use matrix range with matrix-matrix product:
//
ublas_A_sub1 += prod(ublas_C_sub, ublas_D_sub);
vcl_A_sub1 += viennacl::linalg::prod(vcl_C_sub, vcl_D_sub);
//
// Print result matrices:
//
std::cout << "Result ublas: " << ublas_A << std::endl;
std::cout << "Result ViennaCL: " << vcl_A << std::endl;
//
// That's it.
//
std::cout << "!!!! TUTORIAL COMPLETED SUCCESSFULLY !!!!" << std::endl;
return EXIT_SUCCESS;
}
int run_test()
{
//typedef float ScalarType;
typedef boost::numeric::ublas::matrix<ScalarType> MatrixType;
typedef boost::numeric::ublas::vector<ScalarType> VectorType;
typedef viennacl::matrix<ScalarType, T> VCLMatrixType;
typedef viennacl::vector<ScalarType> VCLVectorType;
viennacl::scalar<ScalarType> gpu_pi = ScalarType(3.1415);
std::size_t dim_large = 151;
std::size_t dim_small = 37;
//std::size_t dim_large = 35;
//std::size_t dim_small = 11;
//setup ublas objects:
MatrixType ublas_A(dim_large, dim_large);
for (std::size_t i=0; i<ublas_A.size1(); ++i)
for (std::size_t j=0; j<ublas_A.size2(); ++j)
ublas_A(i,j) = ScalarType((i+1) + (j+1)*(i+1));
MatrixType ublas_B(dim_small, dim_small);
for (std::size_t i=0; i<ublas_B.size1(); ++i)
for (std::size_t j=0; j<ublas_B.size2(); ++j)
ublas_B(i,j) = ScalarType((i+1) + (j+1)*(i+1));
MatrixType ublas_C(dim_large, dim_small);
for (std::size_t i=0; i<ublas_C.size1(); ++i)
for (std::size_t j=0; j<ublas_C.size2(); ++j)
ublas_C(i,j) = ScalarType((j+2) + (j+1)*(i+1));
MatrixType ublas_D(dim_small, dim_large);
for (std::size_t i=0; i<ublas_D.size1(); ++i)
for (std::size_t j=0; j<ublas_D.size2(); ++j)
ublas_D(i,j) = ScalarType((j+2) + (j+1)*(i+1));
boost::numeric::ublas::range ublas_r1(0, dim_small);
boost::numeric::ublas::range ublas_r2(dim_large - dim_small, dim_large);
boost::numeric::ublas::matrix_range<MatrixType> ublas_A_sub1(ublas_A, ublas_r1, ublas_r1);
boost::numeric::ublas::matrix_range<MatrixType> ublas_A_sub2(ublas_A, ublas_r2, ublas_r2);
boost::numeric::ublas::matrix_range<MatrixType> ublas_C_sub(ublas_C, ublas_r1, ublas_r1);
boost::numeric::ublas::matrix_range<MatrixType> ublas_D_sub(ublas_D, ublas_r1, ublas_r1);
//Setup ViennaCL objects
VCLMatrixType vcl_A(dim_large, dim_large);
viennacl::copy(ublas_A, vcl_A);
VCLMatrixType vcl_B(dim_small, dim_small);
viennacl::copy(ublas_B, vcl_B);
VCLMatrixType vcl_C(dim_large, dim_small);
viennacl::copy(ublas_C, vcl_C);
VCLMatrixType vcl_D(dim_small, dim_large);
viennacl::copy(ublas_D, vcl_D);
viennacl::range vcl_r1(0, dim_small);
viennacl::range vcl_r2(dim_large - dim_small, dim_large);
viennacl::matrix_range<VCLMatrixType> vcl_A_sub1(vcl_A, vcl_r1, vcl_r1);
viennacl::matrix_range<VCLMatrixType> vcl_A_sub2(vcl_A, vcl_r2, vcl_r2);
viennacl::matrix_range<VCLMatrixType> vcl_C_sub(vcl_C, vcl_r1, vcl_r1);
viennacl::matrix_range<VCLMatrixType> vcl_D_sub(vcl_D, vcl_r1, vcl_r1);
std::cout << std::endl;
std::cout << "//" << std::endl;
std::cout << "////////// Test: Copy CTOR //////////" << std::endl;
std::cout << "//" << std::endl;
{
std::cout << "Testing matrix created from range... ";
ublas_B = ublas_A_sub1;
VCLMatrixType vcl_temp = vcl_A_sub1;
if (check_for_equality(ublas_B, vcl_temp))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing range created from range... ";
//ublas_A_sub1 = ublas_A_sub1;
VCLMatrixType vcl_ctor_sub1 = vcl_A_sub1; //Note: This is mostly a compilation test only
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
}
std::cout << std::endl;
std::cout << "//" << std::endl;
std::cout << "////////// Test: Assignments //////////" << std::endl;
std::cout << "//" << std::endl;
std::cout << "Testing matrix assigned to range... ";
ublas_A_sub1 = ublas_B;
vcl_A_sub1 = vcl_B;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing range assigned to matrix... ";
ublas_B = ublas_A_sub2;
vcl_B = vcl_A_sub2;
if (check_for_equality(ublas_B, vcl_B))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing range assigned to range... ";
ublas_A_sub1 = ublas_C_sub;
vcl_A_sub1 = vcl_C_sub;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << std::endl;
std::cout << "//" << std::endl;
std::cout << "////////// Test 1: Copy to GPU //////////" << std::endl;
std::cout << "//" << std::endl;
ublas_A_sub1 = ublas_B;
viennacl::copy(ublas_B, vcl_A_sub1);
std::cout << "Testing upper left copy to A... ";
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
ublas_A_sub2 = ublas_B;
viennacl::copy(ublas_B, vcl_A_sub2);
std::cout << "Testing lower right copy to A... ";
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
ublas_C_sub = ublas_B;
viennacl::copy(ublas_B, vcl_C_sub);
std::cout << "Testing upper copy to C... ";
if (check_for_equality(ublas_C, vcl_C))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
ublas_D_sub = ublas_B;
viennacl::copy(ublas_B, vcl_D_sub);
std::cout << "Testing left copy to D... ";
if (check_for_equality(ublas_D, vcl_D))
std::cout << "PASSED!" << std::endl;
else
std::cout << std::endl << "TEST failed!";
std::cout << std::endl;
std::cout << "//" << std::endl;
std::cout << "////////// Test 2: Copy from GPU //////////" << std::endl;
std::cout << "//" << std::endl;
std::cout << "Testing upper left copy to A... ";
if (check_for_equality(ublas_A_sub1, vcl_A_sub1))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing lower right copy to A... ";
if (check_for_equality(ublas_A_sub2, vcl_A_sub2))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing upper copy to C... ";
if (check_for_equality(ublas_C_sub, vcl_C_sub))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Testing left copy to D... ";
if (check_for_equality(ublas_D_sub, vcl_D_sub))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "//" << std::endl;
std::cout << "////////// Test 3: Addition //////////" << std::endl;
std::cout << "//" << std::endl;
viennacl::copy(ublas_A_sub2, vcl_A_sub2);
std::cout << "Inplace add to submatrix: ";
ublas_A_sub2 += ublas_A_sub2;
vcl_A_sub2 += vcl_A_sub2;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Inplace add to matrix: ";
ublas_B += ublas_A_sub2;
vcl_B += vcl_A_sub2;
if (check_for_equality(ublas_B, vcl_B))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Inplace add of matrix: ";
ublas_A_sub2 += ublas_B;
vcl_A_sub2 += vcl_B;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Add to submatrix: ";
ublas_A_sub2 = ublas_A_sub2 + ublas_A_sub2;
vcl_A_sub2 = vcl_A_sub2 + vcl_A_sub2;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Add to matrix: ";
ublas_B = ublas_A_sub2 + ublas_A_sub2;
vcl_B = vcl_A_sub2 + vcl_A_sub2;
if (check_for_equality(ublas_B, vcl_B))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "//" << std::endl;
std::cout << "////////// Test 4: Subtraction //////////" << std::endl;
std::cout << "//" << std::endl;
viennacl::copy(ublas_A_sub2, vcl_A_sub2);
std::cout << "Inplace sub to submatrix: ";
ublas_A_sub2 -= ublas_A_sub2;
vcl_A_sub2 -= vcl_A_sub2;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Inplace sub to matrix: ";
ublas_B -= ublas_A_sub2;
vcl_B -= vcl_A_sub2;
if (check_for_equality(ublas_B, vcl_B))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Inplace sub of matrix: ";
ublas_A_sub2 -= ublas_B;
vcl_A_sub2 -= vcl_B;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Sub from submatrix: ";
ublas_A_sub2 = ublas_A_sub2 - ublas_A_sub2;
vcl_A_sub2 = vcl_A_sub2 - vcl_A_sub2;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Sub from matrix: ";
ublas_B = ublas_A_sub2 - ublas_A_sub2;
vcl_B = vcl_A_sub2 - vcl_A_sub2;
if (check_for_equality(ublas_B, vcl_B))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "//" << std::endl;
std::cout << "////////// Test 5: Scaling //////////" << std::endl;
std::cout << "//" << std::endl;
viennacl::copy(ublas_A, vcl_A);
std::cout << "Multiplication with CPU scalar: ";
ublas_A_sub2 *= ScalarType(3.1415);
vcl_A_sub2 *= ScalarType(3.1415);
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Multiplication with GPU scalar: ";
ublas_A_sub2 *= gpu_pi;
vcl_A_sub2 *= gpu_pi;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Division with CPU scalar: ";
ublas_A_sub2 /= ScalarType(3.1415);
vcl_A_sub2 /= ScalarType(3.1415);
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Division with GPU scalar: ";
ublas_A_sub2 /= gpu_pi;
vcl_A_sub2 /= gpu_pi;
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "//" << std::endl;
std::cout << "////////// Test 6: Matrix-Matrix Products //////////" << std::endl;
std::cout << "//" << std::endl;
std::cout << "Assigned C = A * B: ";
ublas_A_sub1 = prod(ublas_C_sub, ublas_D_sub);
vcl_A_sub1 = viennacl::linalg::prod(vcl_C_sub, vcl_D_sub);
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Assigned C = A^T * B: ";
ublas_A_sub1 = prod(trans(ublas_C_sub), ublas_D_sub);
vcl_A_sub1 = viennacl::linalg::prod(trans(vcl_C_sub), vcl_D_sub);
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Assigned C = A * B^T: ";
ublas_A_sub1 = prod(ublas_C_sub, trans(ublas_D_sub));
vcl_A_sub1 = viennacl::linalg::prod(vcl_C_sub, trans(vcl_D_sub));
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Assigned C = A^T * B^T: ";
ublas_A_sub1 = prod(trans(ublas_C_sub), trans(ublas_D_sub));
vcl_A_sub1 = viennacl::linalg::prod(trans(vcl_C_sub), trans(vcl_D_sub));
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "Inplace add of prod(): ";
ublas_A_sub1 += prod(ublas_C_sub, ublas_D_sub);
vcl_A_sub1 += viennacl::linalg::prod(vcl_C_sub, vcl_D_sub);
if (check_for_equality(ublas_A, vcl_A))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << "//" << std::endl;
std::cout << "////////// Test 7: Matrix-Vector Products //////////" << std::endl;
std::cout << "//" << std::endl;
VectorType ublas_v1(dim_large);
for (std::size_t i=0; i<ublas_v1.size(); ++i)
ublas_v1(i) = i;
boost::numeric::ublas::vector_range<VectorType> ublas_v1_sub(ublas_v1, ublas_r1);
VectorType ublas_v2(dim_large);
for (std::size_t i=0; i<ublas_v2.size(); ++i)
ublas_v2(i) = i - 5;
boost::numeric::ublas::vector_range<VectorType> ublas_v2_sub(ublas_v2, ublas_r1);
VCLVectorType vcl_v1(ublas_v1.size());
viennacl::vector_range<VCLVectorType> vcl_v1_sub(vcl_v1, vcl_r1);
VCLVectorType vcl_v2(ublas_v2.size());
viennacl::vector_range<VCLVectorType> vcl_v2_sub(vcl_v2, vcl_r1);
viennacl::copy(ublas_v1, vcl_v1);
viennacl::copy(ublas_v2, vcl_v2);
viennacl::copy(ublas_A_sub1, vcl_A_sub1);
ublas_v2_sub = prod(ublas_A_sub1, ublas_v1_sub);
vcl_v2_sub = viennacl::linalg::prod(vcl_A_sub1, vcl_v1_sub);
if (check_for_equality_vector(ublas_v2, vcl_v2))
std::cout << "PASSED!" << std::endl;
else
{
std::cout << std::endl << "TEST failed!";
return EXIT_FAILURE;
}
std::cout << std::endl;
std::cout << "----------------------------------------------" << std::endl;
std::cout << std::endl;
return EXIT_SUCCESS;
}