openclWellContributions.hpp

/*
  Copyright 2020 Equinor ASA
 
  This file is part of the Open Porous Media project (OPM).
 
  OPM 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.
 
  OPM 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 OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
 
#ifndef WELLCONTRIBUTIONS_OPENCL_HEADER_INCLUDED
#define WELLCONTRIBUTIONS_OPENCL_HEADER_INCLUDED
 
#include <opm/simulators/linalg/gpubridge/WellContributions.hpp>
 
#include <opm/simulators/linalg/gpubridge/opencl/opencl.hpp>
#include <opm/simulators/linalg/gpubridge/opencl/openclKernels.hpp>
 
#include <memory>
#include <vector>
 
 
namespace Opm {
 
template<class Scalar>
class WellContributionsOCL : public WellContributions<Scalar>
{
public:
    void setOpenCLEnv(cl::Context *context_, cl::CommandQueue *queue_);
 
    void apply_stdwells(cl::Buffer d_x, cl::Buffer d_y);
    void apply_mswells(cl::Buffer d_x, cl::Buffer d_y);
    void apply(cl::Buffer d_x, cl::Buffer d_y);
 
protected:
    void APIalloc() override;
 
    using MatrixType = typename WellContributions<Scalar>::MatrixType;
 
    void APIaddMatrix(MatrixType type, int* colIndices,
                      Scalar* values, unsigned int val_size) override;
 
    cl::Context* context;
    cl::CommandQueue* queue;
    std::vector<cl::Event> events;
 
    std::unique_ptr<cl::Buffer> d_Cnnzs_ocl, d_Dnnzs_ocl, d_Bnnzs_ocl;
    std::unique_ptr<cl::Buffer> d_Ccols_ocl, d_Bcols_ocl;
    std::unique_ptr<cl::Buffer> d_val_pointers_ocl;
 
    std::vector<Scalar> h_x;
    std::vector<Scalar> h_y;
};
 
} // namespace Opm
 
#endif