Sparse-Matrix-Product/main.cpp at master · VannesteFelix/Sparse-Matrix-Product · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#include "component/benchmark.hpp"

int main(int argc, char *argv[])
{
    int benchMarkRun = 0;
    if (argc < 2)
    {
        fprintf(stderr, "Usage: %s [BENCHMARK_RUNS_NUMBER]\n", argv[0]);
        exit(1);
    }
    else
    {
        benchMarkRun = atoi(argv[1]);
    }

    BenchMark<double> myBenchMark_d;
    BenchMark<float> myBenchMark_f;

    std::cout << std::endl;
    std::cout << "----------------------------------------------" << std::endl;
    std::cout << "               Device Info" << std::endl;
    std::cout << "----------------------------------------------" << std::endl;

    #ifdef VIENNACL_WITH_OPENCL
        std::cout << viennacl::ocl::current_device().info() << std::endl;
    #endif

    std::cout << std::endl;

    std::cout << "----------------------------------------------" << std::endl;
    std::cout << "## Benchmark :: Sparse" << std::endl;
    std::cout << "----------------------------------------------" << std::endl;
    std::cout << std::endl;

    //std::cout << "   -------------------------------" << std::endl;
    //std::cout << "   # benchmarking single-precision" << std::endl;
    //std::cout << "   -------------------------------\n" << std::endl;
    //myBenchMark_f.run_benchmark(true);
    //for (int runs=0; runs<benchMarkRun; ++runs)
    //{
    //    myBenchMark_f.run_benchmark(false);
    //}

    #ifdef VIENNACL_WITH_OPENCL
    if ( viennacl::ocl::current_device().double_support() )
    #endif
    {
        std::cout << std::endl;
        std::cout << "   -------------------------------" << std::endl;
        std::cout << "   # benchmarking double-precision" << std::endl;
        std::cout << "   -------------------------------\n" << std::endl;
        myBenchMark_d.run_benchmark(true,true,100);
//        myBenchMark_d.run_benchmark(true,false,1);
        for (int runs=0; runs<benchMarkRun; ++runs)
        {
            myBenchMark_d.run_benchmark(false,true,1);
        }
    }
    return 0;
}