KokkosThread.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#ifdef MOOSE_KOKKOS_SCOPE
#include "KokkosHeader.h"
#endif

#include "MooseTypes.h"

using ThreadID = dof_id_type;

namespace Moose
{
namespace Kokkos
{

class Thread
{
#ifdef MOOSE_KOKKOS_SCOPE
public:
  void resize(std::vector<ThreadID> dims);
  ThreadID size() const { return _size; }
  KOKKOS_FUNCTION ThreadID operator()(ThreadID tid, unsigned int dim) const
  {
    KOKKOS_ASSERT(dim < _dim);

    return (tid / _strides[dim]) % _dims[dim];
  }
#endif

protected:
  ThreadID _size = 0;
  unsigned int _dim = 0;
  ThreadID _dims[10];
  ThreadID _strides[10];
};

} // namespace Kokkos
} // namespace Moose