/*
* Copyright 2009-2019 The VOTCA Development Team (http://www.votca.org)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
#define BOOST_TEST_MAIN
#define BOOST_TEST_MODULE CudaPipeline_test
#include <boost/test/unit_test.hpp>
#include <string>
#include <votca/xtp/cudapipeline.h>
#include <votca/xtp/eigen.h>
using namespace votca::xtp;
BOOST_AUTO_TEST_SUITE(CudaPipeline_test)
BOOST_AUTO_TEST_CASE(right_matrix_multiplication) {
// Call the class to handle GPU resources
CudaPipeline cuda_pip;
// Call matrix multiplication GPU
Eigen::MatrixXd A = Eigen::MatrixXd::Zero(2, 2);
Eigen::MatrixXd B = Eigen::MatrixXd::Zero(3, 2);
Eigen::MatrixXd C = Eigen::MatrixXd::Zero(3, 2);
Eigen::MatrixXd D = Eigen::MatrixXd::Zero(3, 2);
Eigen::MatrixXd X = Eigen::MatrixXd::Zero(3, 2);
Eigen::MatrixXd Y = Eigen::MatrixXd::Zero(3, 2);
Eigen::MatrixXd Z = Eigen::MatrixXd::Zero(3, 2);
// Define matrices
A << 1., 2., 3., 4.;
B << 5., 6., 7., 8., 9., 10.;
C << 9., 10., 11., 12., 13., 14.;
D << 13., 14., 15., 16., 17., 18.;
X << 23., 34., 31., 46., 39., 58.;
Y << 39., 58., 47., 70., 55., 82.;
Z << 55., 82., 63., 94., 71., 106.;
std::vector<Eigen::MatrixXd> tensor{B, C, D};
std::vector<Eigen::MatrixXd> results(3, Eigen::MatrixXd::Zero(3, 2));
CudaMatrix cuma_A{A, cuda_pip.get_stream()};
CudaMatrix cuma_B{3, 2, cuda_pip.get_stream()};
CudaMatrix cuma_C{3, 2, cuda_pip.get_stream()};
for (votca::Index i = 0; i < 3; i++) {
cuma_B.copy_to_gpu(tensor[i]);
cuda_pip.gemm(cuma_B, cuma_A, cuma_C);
results[i] = cuma_C;
}
// Expected results
BOOST_TEST(X.isApprox(results[0]));
BOOST_TEST(Y.isApprox(results[1]));
BOOST_TEST(Z.isApprox(results[2]));
}
BOOST_AUTO_TEST_CASE(wrong_shape_cublas) {
Eigen::MatrixXd A = Eigen::MatrixXd::Random(2, 2);
Eigen::MatrixXd B = Eigen::MatrixXd::Random(5, 5);
CudaPipeline cuda_pip;
CudaMatrix cuma_A{A, cuda_pip.get_stream()};
CudaMatrix cuma_B{B, cuda_pip.get_stream()};
CudaMatrix cuma_C{2, 5, cuda_pip.get_stream()};
BOOST_REQUIRE_THROW(cuda_pip.gemm(cuma_A, cuma_B, cuma_C),
std::runtime_error);
}
BOOST_AUTO_TEST_SUITE_END()