#include <ROOT/RSpan.hxx>
#include <RConfig.h>
#include <DllImport.h>
#include <cassert>
#include <functional>
#include <string>
#include <vector>
Classes | |
class | RooBatchCompute::Config |
Minimal configuration struct to steer the evaluation of a single node with the RooBatchCompute library. More... | |
struct | RooBatchCompute::ReduceNLLOutput |
class | RooBatchCompute::RooBatchComputeInterface |
The interface which should be implemented to provide optimised computation functions for implementations of RooAbsReal::computeBatch(). More... | |
Namespaces | |
namespace | RooBatchCompute |
Namespace for dispatching RooFit computations to various backends. | |
Typedefs | |
typedef std::vector< double > | RooBatchCompute::ArgVector |
typedef const double *__restrict | RooBatchCompute::InputArr |
typedef double *__restrict | RooBatchCompute::RestrictArr |
typedef std::vector< std::span< const double > > | RooBatchCompute::VarVector |
Functions | |
void | RooBatchCompute::compute (Config cfg, Computer comp, RestrictArr output, size_t size, const VarVector &vars) |
void | RooBatchCompute::compute (Config cfg, Computer comp, RestrictArr output, size_t size, const VarVector &vars, ArgVector &extraArgs) |
Architecture | RooBatchCompute::cpuArchitecture () |
std::string | RooBatchCompute::cpuArchitectureName () |
bool | RooBatchCompute::hasCuda () |
void | RooBatchCompute::init () |
Inspect hardware capabilities, and load the optimal library for RooFit computations. | |
ReduceNLLOutput | RooBatchCompute::reduceNLL (Config cfg, std::span< const double > probas, std::span< const double > weights, std::span< const double > offsetProbas) |
double | RooBatchCompute::reduceSum (Config cfg, InputArr input, size_t n) |
Variables | |
R__EXTERN RooBatchComputeInterface * | RooBatchCompute::dispatchCPU = nullptr |
This dispatch pointer points to an implementation of the compute library, provided one has been loaded. | |
R__EXTERN RooBatchComputeInterface * | RooBatchCompute::dispatchCUDA = nullptr |