VectorizedSolver.h

Go to the documentation of this file.

#ifndef VECTORIZEDSOLVER_H_
#define VECTORIZEDSOLVER_H_

#ifdef __cplusplus
#define _USE_MATH_DEFINES
#include "CPUSolver.h"
#include <math.h>
#include <omp.h>
#include <stdlib.h>
#include <mkl.h>
#endif

#define taus(p,e) (taus[(p)*_num_groups + (e)])

#define exponentials(p,e) (exponentials[(p)*_num_groups + (e)])

class VectorizedSolver : public CPUSolver {

protected:

  int _num_vector_lengths;

  FP_PRECISION* _delta_psi;

  FP_PRECISION* _thread_taus;

  FP_PRECISION* _thread_exponentials;

  void initializeExpEvaluator();
  void initializeFluxArrays();
  void initializeSourceArrays();

  void normalizeFluxes();
  void computeFSRSources(int iteration);
  void addSourceToScalarFlux();
  void computeKeff();

  void tallyScalarFlux(segment* curr_segment, int azim_index,
                       FP_PRECISION* track_flux, FP_PRECISION* fsr_flux);
  void transferBoundaryFlux(int track_id, int azim_index, bool direction,
                            FP_PRECISION* track_flux);
  void computeExponentials(segment* curr_segment, FP_PRECISION* exponentials);

public:
  VectorizedSolver(TrackGenerator* track_generator=NULL);
  virtual ~VectorizedSolver();

  int getNumVectorWidths();

  void setFixedSourceByFSR(int fsr_id, int group,
                           FP_PRECISION source);
  void setGeometry(Geometry* geometry);
};


#endif /* VECTORIZEDSOLVER_H_ */