Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feature: rng primitive refactoring #2968

Open
wants to merge 24 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 19 commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
294b2f2
fixes
Alexandr-Solovev Nov 5, 2024
81d7dfe
minor fixes
Alexandr-Solovev Nov 5, 2024
acb6e4c
adding mrg32k3a engine
Alexandr-Solovev Nov 13, 2024
58d98b0
fix fro mrg32k
Alexandr-Solovev Nov 13, 2024
67ed2f6
add philox
Alexandr-Solovev Nov 14, 2024
4941b23
a lot of fixes with rng
Alexandr-Solovev Nov 15, 2024
5ff779f
Merge branch 'oneapi-src:main' into dev/asolovev_rng_primitive_refact…
Alexandr-Solovev Nov 18, 2024
c55bcce
fixes
Alexandr-Solovev Nov 18, 2024
806a74c
clang + fisher yates
Alexandr-Solovev Nov 18, 2024
cc85e37
refactoring
Alexandr-Solovev Nov 19, 2024
852669f
fixes
Alexandr-Solovev Nov 19, 2024
d26465d
Merge branch 'uxlfoundation:main' into dev/asolovev_rng_primitive_ref…
Alexandr-Solovev Dec 5, 2024
6ffe6c5
Merge branch 'uxlfoundation:main' into dev/asolovev_rng_primitive_ref…
Alexandr-Solovev Dec 16, 2024
06f1885
comments fixes
Alexandr-Solovev Dec 16, 2024
72755db
minor fixes
Alexandr-Solovev Dec 16, 2024
76967f3
minor fixes
Alexandr-Solovev Dec 17, 2024
917fa3d
fixes
Alexandr-Solovev Dec 17, 2024
06d9f82
add comments and minor renaming
Alexandr-Solovev Dec 17, 2024
c738f55
Merge branch 'uxlfoundation:main' into dev/asolovev_rng_primitive_ref…
Alexandr-Solovev Dec 19, 2024
42e0020
minor fix
Alexandr-Solovev Dec 20, 2024
85d9f02
fix
Alexandr-Solovev Dec 20, 2024
7393a26
minor fix
Alexandr-Solovev Dec 20, 2024
11b1505
minor fixes
Alexandr-Solovev Jan 9, 2025
ea56291
Merge branch 'uxlfoundation:main' into dev/asolovev_rng_primitive_ref…
Alexandr-Solovev Jan 9, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
183 changes: 183 additions & 0 deletions cpp/daal/include/algorithms/engines/mrg32k3a/mrg32k3a.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,183 @@
/* file: mrg32k3a.h */
/*******************************************************************************
* Copyright contributors to the oneDAL project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*******************************************************************************/

/*
//++
// Implementation of the MRG32k3a engine: a 32-bit combined multiple recursive generator
// with two components of order 3, optimized for batch processing.
//--
*/

#ifndef __MRG32K3A_H__
#define __MRG32K3A_H__

#include "algorithms/engines/mrg32k3a/mrg32k3a_types.h"
#include "algorithms/engines/engine.h"

namespace daal
{
namespace algorithms
{
namespace engines
{
namespace mrg32k3a
{
/**
* @defgroup engines_mrg32k3a_batch Batch
* @ingroup engines_mrg32k3a
* @{
*/
namespace interface1
{
/**
* <a name="DAAL-CLASS-ALGORITHMS__ENGINES__mrg32k3a__BATCHCONTAINER"></a>
* \brief Provides methods to run implementations of the mrg32k3a engine.
* This class is associated with the \ref mrg32k3a::interface1::Batch "mrg32k3a::Batch" class
* and supports the method of mrg32k3a engine computation in the batch processing mode
*
* \tparam algorithmFPType Data type to use in intermediate computations of mrg32k3a engine, double or float
* \tparam method Computation method of the engine, mrg32k3a::Method
* \tparam cpu Version of the cpu-specific implementation of the engine, daal::CpuType
*/
template <typename algorithmFPType, Method method, CpuType cpu>
class BatchContainer : public daal::algorithms::AnalysisContainerIface<batch>
{
public:
/**
* Constructs a container for the mrg32k3a engine with a specified environment
* in the batch processing mode
* \param[in] daalEnv Environment object
*/
BatchContainer(daal::services::Environment::env * daalEnv);
~BatchContainer();
/**
* Computes the result of the mrg32k3a engine in the batch processing mode
*
* \return Status of computations
*/
services::Status compute() DAAL_C11_OVERRIDE;
};

/**
* <a name="DAAL-CLASS-ALGORITHMS__ENGINES__mrg32k3a__BATCH"></a>
* \brief Provides methods for mrg32k3a engine computations in the batch processing mode
*
* \tparam algorithmFPType Data type to use in intermediate computations of mrg32k3a engine, double or float
* \tparam method Computation method of the engine, mrg32k3a::Method
*
* \par Enumerations
* - mrg32k3a::Method Computation methods for the mrg32k3a engine
*
* \par References
* - \ref engines::interface1::Input "engines::Input" class
* - \ref engines::interface1::Result "engines::Result" class
*/
template <typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
class DAAL_EXPORT Batch : public engines::BatchBase
{
public:
typedef engines::BatchBase super;

typedef typename super::InputType InputType;
typedef typename super::ResultType ResultType;

/**
* Creates mrg32k3a engine
* \param[in] seed Initial condition for mrg32k3a engine
*
* \return Pointer to mrg32k3a engine
*/
static services::SharedPtr<Batch<algorithmFPType, method> > create(size_t seed = 777);

/**
* Returns method of the engine
* \return Method of the engine
*/
virtual int getMethod() const DAAL_C11_OVERRIDE { return (int)method; }

/**
* Returns the structure that contains results of mrg32k3a engine
* \return Structure that contains results of mrg32k3a engine
*/
ResultPtr getResult() { return _result; }

/**
* Registers user-allocated memory to store results of mrg32k3a engine
* \param[in] result Structure to store results of mrg32k3a engine
*
* \return Status of computations
*/
services::Status setResult(const ResultPtr & result)
{
DAAL_CHECK(result, services::ErrorNullResult)
_result = result;
_res = _result.get();
return services::Status();
}

/**
* Returns a pointer to the newly allocated mrg32k3a engine
* with a copy of input objects and parameters of this mrg32k3a engine
* \return Pointer to the newly allocated engine
*/
services::SharedPtr<Batch<algorithmFPType, method> > clone() const { return services::SharedPtr<Batch<algorithmFPType, method> >(cloneImpl()); }

/**
* Allocates memory to store the result of the mrg32k3a engine
*
* \return Status of computations
*/
virtual services::Status allocateResult() DAAL_C11_OVERRIDE
{
services::Status s = this->_result->template allocate<algorithmFPType>(&(this->input), NULL, (int)method);
this->_res = this->_result.get();
return s;
}

protected:
Batch(size_t seed = 777) { initialize(); }

Batch(const Batch<algorithmFPType, method> & other) : super(other) { initialize(); }

virtual Batch<algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE { return new Batch<algorithmFPType, method>(*this); }

void initialize()
{
Analysis<batch>::_ac = new __DAAL_ALGORITHM_CONTAINER(batch, BatchContainer, algorithmFPType, method)(&_env);
_in = &input;
_result.reset(new ResultType());
}

private:
ResultPtr _result;

Batch & operator=(const Batch &);
};
typedef services::SharedPtr<Batch<> > mrg32k3aPtr;
typedef services::SharedPtr<const Batch<> > mrg32k3aConstPtr;

} // namespace interface1
using interface1::BatchContainer;
using interface1::Batch;
using interface1::mrg32k3aPtr;
using interface1::mrg32k3aConstPtr;
/** @} */
} // namespace mrg32k3a
} // namespace engines
} // namespace algorithms
} // namespace daal
#endif
65 changes: 65 additions & 0 deletions cpp/daal/include/algorithms/engines/mrg32k3a/mrg32k3a_types.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
/* file: mrg32k3a_types.h */
/*******************************************************************************
* Copyright contributors to the oneDAL project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*******************************************************************************/

/*
//++
// Implementation of the MRG32k3a engine: a 32-bit combined multiple recursive generator
// with two components of order 3, optimized for batch processing.
//--
*/

#ifndef __MRG32K3A_TYPES_H__
#define __MRG32K3A_TYPES_H__

#include "algorithms/algorithm.h"
#include "services/daal_defines.h"
#include "data_management/data/numeric_table.h"
#include "data_management/data/homogen_numeric_table.h"

namespace daal
{
namespace algorithms
{
namespace engines
{
/**
* @defgroup engines_mrg32k3a mrg32k3a Engine
* \copydoc daal::algorithms::engines::mrg32k3a
* @ingroup engines
* @{
*/
/**
* \brief Contains classes for mrg32k3a engine
*/
namespace mrg32k3a
{
/**
* <a name="DAAL-ENUM-ALGORITHMS__ENGINES__mrg32k3a__METHOD"></a>
* Available methods to compute mrg32k3a engine
*/
enum Method
{
defaultDense = 0 /*!< Default: performance-oriented method. */
};

} // namespace mrg32k3a
/** @} */
} // namespace engines
} // namespace algorithms
} // namespace daal

#endif
Loading
Loading