src/cudaSolvers/cudaSolver/cudaTypes.H

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | foam-extend: Open Source CFD
   4    \\    /   O peration     | Version:     3.2
   5     \\  /    A nd           | Web:         http://www.foam-extend.org
   6      \\/     M anipulation  | For copyright notice see file Copyright
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of foam-extend.
  10
  11     foam-extend is free software: you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by the
  13     Free Software Foundation, either version 3 of the License, or (at your
  14     option) any later version.
  15
  16     foam-extend is distributed in the hope that it will be useful, but
  17     WITHOUT ANY WARRANTY; without even the implied warranty of
  18     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19     General Public License for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 Class
  25     Foam::cudaSolver
  26
  27 Description
  28     Base class for GPU based solvers
  29
  30 SourceFiles
  31     cudaSolver.C
  32
  33 Author
  34     Dominik Christ, Wikki Ltd.
  35     Based on Cufflink library by Daniel P. Combest
  36
  37 \*---------------------------------------------------------------------------*/
  38 #ifndef cudaTypes_H
  39 #define cudaTypes_H
  40
  41 #if defined(WM_SP)
  42
  43 #   define MPI_SCALAR MPI_FLOAT
  44     typedef float ValueType;
  45
  46 #elif defined(WM_DP)
  47
  48 #   define MPI_SCALAR MPI_DOUBLE
  49     typedef double ValueType;
  50
  51 #elif defined(WM_LDP)
  52
  53 #   error Long double not supported in CUDA
  54
  55 #endif
  56
  57 #include <cusp/coo_matrix.h>
  58 #include <cusp/memory.h>
  59 #include <thrust/host_vector.h>
  60
  61 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  62
  63 typedef cusp::device_memory MemorySpace;
  64 typedef cusp::host_memory hostMemorySpace;
  65 typedef int IndexType;
  66
  67 // used to prevent floating point exception
  68 const ValueType SMALL = 1e-20;
  69
  70
  71 struct cuspEquationSystem
  72 {
  73     cusp::coo_matrix<IndexType,ValueType,hostMemorySpace> A;
  74     cusp::array1d< ValueType, hostMemorySpace> X;
  75     cusp::array1d< ValueType, hostMemorySpace> B;
  76     int nCells;
  77     int nFaces;
  78 };
  79
  80
  81 struct cudaSolverPerformance
  82 {
  83     int minIter;
  84     int maxIter;
  85     double relTol;
  86     double tol;
  87
  88     int nIterations;
  89     double iRes;
  90     double fRes;
  91     bool converged;
  92     bool singular;
  93
  94     bool debugCusp;
  95 };
  96
  97
  98 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  99
 100 #endif
 101
 102 // ************************************************************************* //