mirror of
https://github.com/triqs/dft_tools
synced 2024-12-26 22:33:48 +01:00
e6234ed3d5
- the conj_ function was not written properly (T is a ref, the trait is_complex was not returning true). - a simpler version is clearly better !
76 lines
2.2 KiB
C++
76 lines
2.2 KiB
C++
/*******************************************************************************
|
|
*
|
|
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
*
|
|
* Copyright (C) 2011 by O. Parcollet
|
|
*
|
|
* TRIQS is free software: you can redistribute it and/or modify it under the
|
|
* terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* TRIQS is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
******************************************************************************/
|
|
#include "./common.hpp"
|
|
#include <iostream>
|
|
|
|
#include <triqs/arrays/array.hpp>
|
|
#include <triqs/arrays/vector.hpp>
|
|
#include <triqs/arrays/matrix.hpp>
|
|
#include <triqs/arrays/blas_lapack/dot.hpp>
|
|
#include <triqs/arrays/asserts.hpp>
|
|
|
|
using namespace triqs::arrays;
|
|
|
|
template<typename Vd, typename Vi>
|
|
void test() {
|
|
Vd a(2),aa(2),c(2) ;a()=2.0; c() = 1;
|
|
Vi b(2);b()=3;
|
|
std::cerr << blas::dot<false>(a,b) << std::endl;
|
|
|
|
aa = 2*a;
|
|
|
|
assert_close( dot(a,b), 12);
|
|
assert_close( dot(aa,a), 16);
|
|
assert_close( dot(aa,b), 24);
|
|
assert_close( dot(aa-a,b), 12);
|
|
|
|
std::cerr << dot(aa,a) << std::endl ;
|
|
std::cerr << dot(aa,b) << std::endl ;
|
|
std::cerr << dot(aa-a, b) << std::endl;
|
|
std::cerr<< "---------------------"<<std::endl;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
test<vector<double> , vector<int> >();
|
|
|
|
/// Added by I. Krivenko, #122
|
|
/// test the complex version, specially with the zdotu workaround on Os X.
|
|
vector<std::complex<double>> v(2);
|
|
v(0) = 0;
|
|
v(1) = {0, 1};
|
|
|
|
std::cerr << v << std::endl;
|
|
std::cerr << blas::dot<false>(v, v) << std::endl;
|
|
std::cerr << blas::dot<true>(v, v) << std::endl;
|
|
assert_close( dot(v,v), -1);
|
|
assert_close( dotc(v,v), 1);
|
|
|
|
// does not work for array just because of .size() vs .shape(0)...
|
|
//test<array<double,1> , array<int,1> >();
|
|
//test<array<double,1> , vector<int> >();
|
|
//test<vector<double> , array<int,1> >();
|
|
|
|
}
|
|
|
|
|
|
|