56 lines
2.1 KiB
C++
56 lines
2.1 KiB
C++
// ======================================================================== //
|
|
// Copyright 2009-2019 Intel Corporation //
|
|
// //
|
|
// 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. //
|
|
// ======================================================================== //
|
|
|
|
#include "tasking.h"
|
|
|
|
namespace oidn {
|
|
|
|
// --------------------------------------------------------------------------
|
|
// PinningObserver
|
|
// --------------------------------------------------------------------------
|
|
|
|
PinningObserver::PinningObserver(const std::shared_ptr<ThreadAffinity>& affinity)
|
|
: affinity(affinity)
|
|
{
|
|
observe(true);
|
|
}
|
|
|
|
PinningObserver::PinningObserver(const std::shared_ptr<ThreadAffinity>& affinity, tbb::task_arena& arena)
|
|
: tbb::task_scheduler_observer(arena),
|
|
affinity(affinity)
|
|
{
|
|
observe(true);
|
|
}
|
|
|
|
PinningObserver::~PinningObserver()
|
|
{
|
|
observe(false);
|
|
}
|
|
|
|
void PinningObserver::on_scheduler_entry(bool isWorker)
|
|
{
|
|
const int threadIndex = tbb::this_task_arena::current_thread_index();
|
|
affinity->set(threadIndex);
|
|
}
|
|
|
|
void PinningObserver::on_scheduler_exit(bool isWorker)
|
|
{
|
|
const int threadIndex = tbb::this_task_arena::current_thread_index();
|
|
affinity->restore(threadIndex);
|
|
}
|
|
|
|
} // namespace oidn
|