zswizzle

Source: tests/dense/swizzle.cpp

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

// Containers.
#include <myramath/utility/Number.h>
#include <myramath/dense/Matrix.h>
#include <myramath/dense/MatrixRange.h>

// Algorithms.
#include <myramath/dense/gemm.h>
#include <myramath/dense/swizzle.h>
#include <myramath/dense/frobenius.h>

// Real solver type for A.
#include <myramath/dense/LUSolver.h>

// Reporting.
#include <tests/myratest.h>

using namespace myra;

namespace {

template<class Precision> void test(int I, int J, Precision tolerance)
  {
  myra::out() << typestring<Precision>() << std::endl;
  // Make random real A, and generate a solver for it.
  const auto A = Matrix<Precision>::random(I,I);
  LUSolver<Precision> solver(A);
  // Make random complex X, and multiply it into B=A*X using make_complex(A)
  typedef std::complex<Precision> Number;
  auto X = Matrix<Number>::random(I,J);
  Matrix<Number> B = make_complex(A)*X;
  // Use solver.solve() and swizzle()/unswizzle() to solve A*X=B.
  auto RI = swizzle(B);
  solver.solve(RI.first);
  solver.solve(RI.second);
  unswizzle(B);
  // Verify B=X.
  Precision error = frobenius(X-B)/frobenius(X);
  myra::out() << "inv(A) [real] * B [complex] - X [complex] == " << error << std::endl;
  REQUIRE(error < tolerance);
  }

} // namespace

ADD_TEST("cswizzle","[dense][solver]")
  { test<float>(53,21,1.0e-3f); }

ADD_TEST("zswizzle","[dense][solver]")
  { test<double>(48,18,1.0e-10); }

Results: [PASS]

double
inv(A) [real] * B [complex] - X [complex] == 5.30887e-15

Go back to Summary of /test programs.