TestMatrixArithmeticCuda.cxx

Go to the documentation of this file.

// @(#)root/tmva $Id$
// Author: Simon Pfreundschuh

/*************************************************************************
 * Copyright (C) 2016, Simon Pfreundschuh
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

////////////////////////////////////////////////////////////////////
// Concrete instantiation of the generic backpropagation test for //
// CUDA architectures.                                            //
////////////////////////////////////////////////////////////////////

#include <iostream>
#include "TMVA/DNN/Architectures/Cuda.h"
#include "TestMatrixArithmetic.h"

using namespace TMVA::DNN;

int main()
{
    std::cout << "Testing CUDA matrix arithmetic (double):" << std::endl;

    Double_t error = testMultiplication<TCuda<Double_t>>(10);
    std::cout << "Multiplication: " << "Max. rel. error: " << error << std::endl;
    if (error > 1e-3)
        return 1;

    error = testSumColumns<TCuda<Double_t>>(1);
    std::cout << "Column Sum:     " << "Max. rel. error: " << error << std::endl;
    if (error > 1e-3)
        return 1;

    std::cout << "Testing CUDA matrix arithmetic (float):" << std::endl;

    error = testMultiplication<TCuda<Real_t>>(10);
    std::cout << "Multiplication: " << "Max. rel. error: " << error << std::endl;
    if (error > 1)
        return 1;

    error = testSumColumns<TCuda<Real_t>>(1);
    std::cout << "Column Sum:     " << "Max. rel. error: " << error << std::endl;
    if (error > 1e-3)
        return 1;
}