From 3ead57fc38871246327687bce4772714f952bc4b Mon Sep 17 00:00:00 2001 From: Wolfgang Bangerth Date: Thu, 18 Jan 2018 15:45:56 -0700 Subject: [PATCH] Add a test for LinearAlgebra::distributed::Vector::add_and_dot(). --- .../la_parallel_vector_add_and_dot_complex.cc | 78 +++++++++++++++++++ ...parallel_vector_add_and_dot_complex.output | 22 ++++++ 2 files changed, 100 insertions(+) create mode 100644 tests/lac/la_parallel_vector_add_and_dot_complex.cc create mode 100644 tests/lac/la_parallel_vector_add_and_dot_complex.output diff --git a/tests/lac/la_parallel_vector_add_and_dot_complex.cc b/tests/lac/la_parallel_vector_add_and_dot_complex.cc new file mode 100644 index 0000000000..ae740d55a2 --- /dev/null +++ b/tests/lac/la_parallel_vector_add_and_dot_complex.cc @@ -0,0 +1,78 @@ +// --------------------------------------------------------------------- +// +// Copyright (C) 2012 - 2015 by the deal.II authors +// +// This file is part of the deal.II library. +// +// The deal.II library is free software; you can use it, redistribute +// it, and/or modify it under the terms of the GNU Lesser General +// Public License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// The full text of the license can be found in the file LICENSE at +// the top level of the deal.II distribution. +// +// --------------------------------------------------------------------- + + +// check that LinearAlgebra::distributed::Vector::add_and_dot works correctly for complex-valued +// vectors + +#include "../tests.h" +#include + + + + +template +void check () +{ + for (unsigned int test=0; test<5; ++test) + { + const unsigned int size = 17 + test*1101; + const IndexSet complete_set = complete_index_set(size); + LinearAlgebra::distributed::Vector> v1 (complete_set, MPI_COMM_SELF), v2(complete_set, MPI_COMM_SELF), v3(complete_set, MPI_COMM_SELF), check(complete_set, MPI_COMM_SELF); + for (unsigned int i=0; i(0.1 + 0.005 * i, 1.234 + 12 * i); + v2(i) = std::complex(-5.2 + 0.18 * i, 42.4242 + 42 * i); + v3(i) = std::complex(3.14159 + 2.7183/(1.+i), 13.); + } + check = v1; + const std::complex factor = std::complex(0.01432); + + // do things by hand once + v1.add(factor, v2); + const std::complex prod = v1 * v3; + + // then do it a second time with the add_and_dot function + const std::complex prod_check = check.add_and_dot(factor, v2, v3); + if (test == 0 && std::is_same::value) + { + deallog << "Vector add reference: "; + v1.print(deallog.get_file_stream()); + deallog << std::endl; + deallog << "Vector check reference: "; + check.print(deallog.get_file_stream()); + deallog << std::endl; + } + + deallog << "Add and dot should be " << prod/static_cast(size) + << ", is " << prod_check/static_cast(size) + << std::endl; + } +} + + +int main() +{ + std::ofstream logfile("output"); + deallog << std::fixed; + deallog << std::setprecision(2); + deallog.attach(logfile); + + check(); + check(); + deallog << "OK" << std::endl; +} + + diff --git a/tests/lac/la_parallel_vector_add_and_dot_complex.output b/tests/lac/la_parallel_vector_add_and_dot_complex.output new file mode 100644 index 0000000000..0dc50c403e --- /dev/null +++ b/tests/lac/la_parallel_vector_add_and_dot_complex.output @@ -0,0 +1,22 @@ + +DEAL::Add and dot should be (1334.79,349.71), is (1334.79,349.71) +DEAL::Add and dot should be (91530.09,22094.70), is (91530.09,22094.70) +DEAL::Add and dot should be (181725.41,43834.06), is (181725.41,43834.06) +DEAL::Add and dot should be (271920.72,65573.37), is (271920.72,65573.37) +DEAL::Add and dot should be (362116.03,87312.66), is (362116.03,87312.66) +Process #0 +Local range: [0, 17), global size: 17 +Vector data: +(2.554e-02,1.842e+00) (3.311e-02,1.444e+01) (4.069e-02,2.704e+01) (4.827e-02,3.965e+01) (5.585e-02,5.225e+01) (6.342e-02,6.485e+01) (7.100e-02,7.745e+01) (7.858e-02,9.005e+01) (8.616e-02,1.027e+02) (9.373e-02,1.153e+02) (1.013e-01,1.279e+02) (1.089e-01,1.405e+02) (1.165e-01,1.531e+02) (1.240e-01,1.657e+02) (1.316e-01,1.783e+02) (1.392e-01,1.909e+02) (1.468e-01,2.035e+02) +DEAL::Vector add reference: +Process #0 +Local range: [0, 17), global size: 17 +Vector data: +(2.554e-02,1.842e+00) (3.311e-02,1.444e+01) (4.069e-02,2.704e+01) (4.827e-02,3.965e+01) (5.585e-02,5.225e+01) (6.342e-02,6.485e+01) (7.100e-02,7.745e+01) (7.858e-02,9.005e+01) (8.616e-02,1.027e+02) (9.373e-02,1.153e+02) (1.013e-01,1.279e+02) (1.089e-01,1.405e+02) (1.165e-01,1.531e+02) (1.240e-01,1.657e+02) (1.316e-01,1.783e+02) (1.392e-01,1.909e+02) (1.468e-01,2.035e+02) +DEAL::Vector check reference: +DEAL::Add and dot should be (1334.79,349.71), is (1334.79,349.71) +DEAL::Add and dot should be (91530.09,22094.70), is (91530.09,22094.70) +DEAL::Add and dot should be (181725.40,43834.06), is (181725.40,43834.06) +DEAL::Add and dot should be (271920.71,65573.37), is (271920.71,65573.37) +DEAL::Add and dot should be (362116.02,87312.66), is (362116.02,87312.66) +DEAL::OK -- 2.39.5