RecoTracker/MkFitCore/src/Matrix.h - master - githubmirror/cmssw

RecoTracker/MkFitCore/src/Matrix.h
90 строк3 KB
Mario Masciovecchio
Intermediate code-format
15 окт 2024, 15:09
15 окт 2024, 15:097f57653
#ifndef RecoTracker_MkFitCore_src_Matrix_h
#define RecoTracker_MkFitCore_src_Matrix_h

#include "RecoTracker/MkFitCore/interface/Config.h"
#include "RecoTracker/MkFitCore/interface/MatrixSTypes.h"

namespace mkfit {

  inline float hipo(float x, float y) { return std::sqrt(x * x + y * y); }

  inline float hipo_sqr(float x, float y) { return x * x + y * y; }

  inline void sincos4(const float x, float& sin, float& cos) {
    // Had this writen with explicit division by factorial.
    // The *whole* fitting test ran like 2.5% slower on MIC, sigh.

    const float x2 = x * x;
    cos = 1.f - 0.5f * x2 + 0.04166667f * x2 * x2;
    sin = x - 0.16666667f * x * x2;
  }
}  // end namespace mkfit

//==============================================================================

// Matriplex dimensions and typedefs

// Provide fast_xyzz() Matriplex methods and operators using VDT.
#define MPLEX_VDT
// Define the following to have fast_xyzz() functions actually call std:: stuff.
// #define MPLEX_VDT_USE_STD

#include "Matriplex/MatriplexSym.h"

#ifndef MPT_SIZE
#if defined(__AVX512F__)
#define MPT_SIZE 16
#elif defined(__AVX__) || defined(__AVX2__)
#define MPT_SIZE 8
#elif defined(__SSE3__)
#define MPT_SIZE 4
#else
#define MPT_SIZE 8
#endif
#endif

namespace mkfit {

  constexpr Matriplex::idx_t NN = MPT_SIZE;  // "Length" of MPlex.

  constexpr Matriplex::idx_t LL = 6;  // Dimension of large/long  MPlex entities
  constexpr Matriplex::idx_t HH = 3;  // Dimension of small/short MPlex entities

  typedef Matriplex::Matriplex<float, LL, LL, NN> MPlexLL;
  typedef Matriplex::Matriplex<float, LL, 1, NN> MPlexLV;
  typedef Matriplex::MatriplexSym<float, LL, NN> MPlexLS;

  typedef Matriplex::Matriplex<float, HH, HH, NN> MPlexHH;
  typedef Matriplex::Matriplex<float, HH, 1, NN> MPlexHV;
  typedef Matriplex::MatriplexSym<float, HH, NN> MPlexHS;

  typedef Matriplex::Matriplex<float, 5, 1, NN> MPlex5V;
  typedef Matriplex::MatriplexSym<float, 5, NN> MPlex5S;

  typedef Matriplex::Matriplex<float, 5, 5, NN> MPlex55;
  typedef Matriplex::Matriplex<float, 5, 6, NN> MPlex56;
  typedef Matriplex::Matriplex<float, 6, 5, NN> MPlex65;

  typedef Matriplex::Matriplex<float, 2, 2, NN> MPlex22;
  typedef Matriplex::Matriplex<float, 2, 1, NN> MPlex2V;
  typedef Matriplex::MatriplexSym<float, 2, NN> MPlex2S;

  typedef Matriplex::Matriplex<float, LL, HH, NN> MPlexLH;
  typedef Matriplex::Matriplex<float, HH, LL, NN> MPlexHL;

  typedef Matriplex::Matriplex<float, 5, 2, NN> MPlex52;
  typedef Matriplex::Matriplex<float, LL, 2, NN> MPlexL2;
  typedef Matriplex::Matriplex<float, HH, 2, NN> MPlexH2;
  typedef Matriplex::Matriplex<float, 2, HH, NN> MPlex2H;

  typedef Matriplex::Matriplex<float, 1, 1, NN> MPlexQF;
  typedef Matriplex::Matriplex<int, 1, 1, NN> MPlexQI;
  typedef Matriplex::Matriplex<unsigned int, 1, 1, NN> MPlexQUI;
  typedef Matriplex::Matriplex<short, 1, 1, NN> MPlexQH;
  typedef Matriplex::Matriplex<unsigned short, 1, 1, NN> MPlexQUH;

  typedef Matriplex::Matriplex<bool, 1, 1, NN> MPlexQB;

}  // end namespace mkfit

#endif