Skip to content

Commit

Permalink
Add debug prints
Browse files Browse the repository at this point in the history
  • Loading branch information
avolkov-intel committed Jul 10, 2024
1 parent d03fe03 commit 6274d7d
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 1 deletion.
6 changes: 6 additions & 0 deletions cpp/daal/src/services/env_detect.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
#include "src/externals/service_service.h"
#include "src/threading/threading.h"
#include "services/error_indexes.h"
#include <iostream>

#include "src/services/service_topo.h"
#include "src/threading/service_thread_pinner.h"
Expand Down Expand Up @@ -127,6 +128,7 @@ DAAL_EXPORT void daal::services::Environment::setDynamicLibraryThreadingTypeOnWi

DAAL_EXPORT daal::services::Environment::Environment()
{
std::cout << "Env constructor" << std::endl;
_env.cpuid_init_flag = false;
_env.cpuid = -1;
this->setDefaultExecutionContext(internal::CpuExecutionContext());
Expand All @@ -138,6 +140,7 @@ DAAL_EXPORT daal::services::Environment::Environment(const Environment & e) : da
DAAL_EXPORT void daal::services::Environment::initNumberOfThreads()
{
if (isInit) return;
std::cout << "Inside init number of threads" << std::endl;

/* if HT enabled - set _numThreads to physical cores num */
if (daal::internal::ServiceInst::serv_get_ht())
Expand All @@ -149,18 +152,21 @@ DAAL_EXPORT void daal::services::Environment::initNumberOfThreads()

if (ncores > 0)
{
std::cout << "(ht enabled) init with " << ncores << std::endl;
daal::services::Environment::setNumberOfThreads(ncores);
}
}
else
{
std::cout << "(ht disabled) init with " << _daal_threader_get_max_threads() << std::endl;
daal::services::Environment::setNumberOfThreads(_daal_threader_get_max_threads());
}
isInit = true;
}

DAAL_EXPORT daal::services::Environment::~Environment()
{
std::cout << "Env destructor" << std::endl;
daal::services::daal_free_buffers();
}

Expand Down
11 changes: 10 additions & 1 deletion cpp/daal/src/threading/threading.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -35,24 +35,31 @@
#include <tbb/global_control.h>
#include <tbb/task_arena.h>
#include "services/daal_atomic_int.h"
#include <iostream>

#if defined(TBB_INTERFACE_VERSION) && TBB_INTERFACE_VERSION >= 12002
#include <tbb/task.h>
#endif

namespace daal
{
ThreaderEnvironment::ThreaderEnvironment() : _numberOfThreads(1), _taskArena(nullptr) {}
ThreaderEnvironment::ThreaderEnvironment() : _numberOfThreads(1), _taskArena(nullptr)
{
std::cout << "threader env constructor" << std::endl;
}
ThreaderEnvironment::~ThreaderEnvironment()
{
std::cout << "threader env destructor" << std::endl;
if (_taskArena)
{
std::cout << "delete task arena" << std::endl;
delete reinterpret_cast<tbb::task_arena *>(_taskArena);
_taskArena = nullptr;
}
}
void ThreaderEnvironment::setNumberOfThreads(size_t value)
{
std::cout << "set number of threads from " << _numberOfThreads << " to " << value << std::endl;
if (_taskArena)
{
delete reinterpret_cast<tbb::task_arena *>(_taskArena);
Expand Down Expand Up @@ -80,12 +87,14 @@ DAAL_EXPORT void _threaded_scalable_free(void * ptr)

DAAL_EXPORT size_t _setNumberOfThreads(const size_t numThreads)
{
std::cout << "set nthreads " << numThreads << std::endl;
static tbb::spin_mutex mt;
tbb::spin_mutex::scoped_lock lock(mt);
if (numThreads > 1)
{
const size_t maxNumThreads = _daal_threader_get_max_threads();
const size_t limitedNumThreads = numThreads < maxNumThreads ? numThreads : maxNumThreads;
std::cout << "Set number of threads to " << limitedNumThreads << " max(" << maxNumThreads << ")" << std::endl;
daal::threader_env()->setNumberOfThreads(limitedNumThreads);
return limitedNumThreads;
}
Expand Down

0 comments on commit 6274d7d

Please sign in to comment.