Uses OMP C++ interface rather than grabbing potentially non-existent …

…environment variable
mach3-software · Aug 1, 2024 · cdf64d1 · cdf64d1
1 parent 8ac572f
commit cdf64d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samplePDF/samplePDFFDBase.cpp b/samplePDF/samplePDFFDBase.cpp
@@ -859,7 +859,7 @@ void samplePDFFDBase::SetupOscCalc(double PathLength, double Density)
 //if we just want to use CUDAProb3 CPU then setup BeamCpuPropagator object
 #if defined (MULTITHREAD)
 //if we want to multithread then get number of threads from OMP_NUM_THREADS env variable
-    MCSamples[iSample].Oscillator = new cudaprob3::BeamCpuPropagator<double>(MCSamples[iSample].nEvents, std::atoi(std::getenv("OMP_NUM_THREADS")));
+    MCSamples[iSample].Oscillator = new cudaprob3::BeamCpuPropagator<double>(MCSamples[iSample].nEvents, omp_get_max_threads()));
   MCSamples[iSample].Oscillator->setPathLength(PathLength);
   MCSamples[iSample].Oscillator->setDensity(Density);
 #else