e2iphi.h

Go to the documentation of this file.

//////////////////////////////////////////////////////////////////////////////////////
// This file is distributed under the University of Illinois/NCSA Open Source License.
// See LICENSE file in top directory for details.
//
// Copyright (c) 2016 Jeongnim Kim and QMCPACK developers.
//
// File developed by: Ken Esler, kpesler@gmail.com, University of Illinois at Urbana-Champaign
//                    Jeremy McMinnis, jmcminis@gmail.com, University of Illinois at Urbana-Champaign
//                    Jeongnim Kim, jeongnim.kim@gmail.com, University of Illinois at Urbana-Champaign
//                    Mark A. Berrill, berrillma@ornl.gov, Oak Ridge National Laboratory
//                    Mark Dewing, markdewing@gmail.com, University of Illinois at Urbana-Champaign
//                    Paul R. C. Kent, kentpr@ornl.gov, Oak Ridge National Laboratory
//
// File created by: Ken Esler, kpesler@gmail.com, University of Illinois at Urbana-Champaign
//////////////////////////////////////////////////////////////////////////////////////


// Supported libraries and defines:
//
// HAVE_MKL_VML : Intel MKL VML, include with MKL
// HAVE_MASSV   : IBM MASSV library

#ifndef QMCPLUSPLUS_E2IPHI_H
#define QMCPLUSPLUS_E2IPHI_H

#include <config.h>
#include <vector>
#include <complex>
#include "CPU/math.hpp"

#if defined(HAVE_MASSV)
#include <massv.h>
inline void eval_e2iphi(int n, double* restrict phi, double* restrict c, double* restrict s) { vsincos(s, c, phi, &n); }
inline void eval_e2iphi(int n, float* restrict phi, float* restrict c, float* restrict s) { vssincos(s, c, phi, &n); }

inline void eval_e2iphi(int n, double* restrict phi, std::complex<double>* restrict z)
{
  vcosisin((double _Complex*)z, phi, &n);
}

inline void eval_e2iphi(int n, float* restrict phi, std::complex<float>* restrict z)
{
  vscosisin((float _Complex*)z, phi, &n);
}
#elif defined(HAVE_MKL_VML)
#include <mkl_vml_functions.h>
inline void eval_e2iphi(int n, double* restrict phi, double* restrict c, double* restrict s) { vdSinCos(n, phi, s, c); }

inline void eval_e2iphi(int n, float* restrict phi, float* restrict c, float* restrict s) { vsSinCos(n, phi, s, c); }
inline void eval_e2iphi(int n, const double* restrict phi, std::complex<double>* restrict z)
{
  vzCIS(n, phi, (MKL_Complex16*)(z));
}

inline void eval_e2iphi(int n, const float* restrict phi, std::complex<float>* restrict z)
{
  vcCIS(n, phi, (MKL_Complex8*)(z));
}
#else /* generic case */
template<typename T>
inline void eval_e2iphi(int n, const T* restrict phi, T* restrict phase_r, T* restrict phase_i)
{
  for (int i = 0; i < n; i++)
    qmcplusplus::sincos(phi[i], phase_i + i, phase_r + i);
}
template<typename T>
inline void eval_e2iphi(int n, const T* restrict phi, std::complex<T>* restrict z)
{
  T s, c;
  for (int i = 0; i < n; i++)
  {
    qmcplusplus::sincos(phi[i], &s, &c);
    z[i] = std::complex<T>(c, s);
  }
}
#endif

template<typename T>
inline void eval_e2iphi(std::vector<T>& phi, std::vector<std::complex<T>>& z)
{
  eval_e2iphi(phi.size(), &phi[0], &z[0]);
}

#endif