Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
RooBatchCompute::Config Class Reference

Minimal configuration struct to steer the evaluation of a single node with the RooBatchCompute library.

Definition at line 54 of file RooBatchCompute.h.

Public Member Functions

CudaInterface::CudaStreamcudaStream () const
 
int nThreads () const
 
void setCudaStream (CudaInterface::CudaStream *cudaStream)
 
void setNThreads (int nThreads)
 Number of threads to use for CPU batch computations and reductions.
 
bool useCuda () const
 

Private Attributes

CudaInterface::CudaStream_cudaStream = nullptr
 
int _nThreads = 1
 

#include </github/home/ROOT-CI/src/roofit/batchcompute/res/RooBatchCompute.h>

Member Function Documentation

◆ cudaStream()

CudaInterface::CudaStream * RooBatchCompute::Config::cudaStream ( ) const
inline

Definition at line 58 of file RooBatchCompute.h.

◆ nThreads()

int RooBatchCompute::Config::nThreads ( ) const
inline

Definition at line 65 of file RooBatchCompute.h.

◆ setCudaStream()

void RooBatchCompute::Config::setCudaStream ( CudaInterface::CudaStream * cudaStream)
inline

Definition at line 57 of file RooBatchCompute.h.

◆ setNThreads()

void RooBatchCompute::Config::setNThreads ( int nThreads)
inline

Number of threads to use for CPU batch computations and reductions.

Values smaller than two mean single-threaded evaluation. The CPU backend may still evaluate small batches single-threaded to avoid scheduling overhead.

Definition at line 64 of file RooBatchCompute.h.

◆ useCuda()

bool RooBatchCompute::Config::useCuda ( ) const
inline

Definition at line 56 of file RooBatchCompute.h.

Member Data Documentation

◆ _cudaStream

CudaInterface::CudaStream* RooBatchCompute::Config::_cudaStream = nullptr
private

Definition at line 68 of file RooBatchCompute.h.

◆ _nThreads

int RooBatchCompute::Config::_nThreads = 1
private

Definition at line 69 of file RooBatchCompute.h.

  • roofit/batchcompute/res/RooBatchCompute.h