8f43927575
examples/ is based on KasperskyLab/knp at 55470ed6 with local changes to the mnist-learn altai/blifat models, cuda-backend CMake and model_loader. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
132 lines
3.1 KiB
C++
132 lines
3.1 KiB
C++
/**
|
|
* @file cpu.cpp
|
|
* @brief CPU device class implementation.
|
|
* @kaspersky_support Artiom N.
|
|
* @date 30.01.2023
|
|
* @license Apache 2.0
|
|
* @copyright © 2024 AO Kaspersky Lab
|
|
*
|
|
* 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 <knp/devices/cpu.h>
|
|
|
|
#if defined(__clang__)
|
|
# pragma clang diagnostic push
|
|
# pragma clang diagnostic ignored "-Wdocumentation"
|
|
#endif
|
|
|
|
#include <cpucounters.h>
|
|
|
|
#if defined(__clang__)
|
|
# pragma clang diagnostic pop
|
|
#endif
|
|
|
|
#include <spdlog/spdlog.h>
|
|
|
|
#include <exception>
|
|
|
|
#include <boost/uuid/name_generator.hpp>
|
|
|
|
#include "x86_cpu_power.h"
|
|
|
|
|
|
namespace knp::devices::cpu
|
|
{
|
|
|
|
static constexpr const char* ns_uid = "0000-0000-0000-0000";
|
|
|
|
|
|
CPU::CPU(uint32_t cpu_num) : cpu_num_(cpu_num), power_meter_{std::make_unique<CpuPower>(cpu_num)}
|
|
{
|
|
auto pcm_instance = pcm::PCM::getInstance();
|
|
const pcm::PCM::ErrorCode status = pcm_instance->program(pcm::PCM::DEFAULT_EVENTS, nullptr, true, ::getpid());
|
|
check_pcm_status(status);
|
|
|
|
cpu_name_ = pcm_instance->getCPUBrandString() + " " + pcm_instance->getCPUFamilyModelString() + " " +
|
|
std::to_string(cpu_num);
|
|
Device::base_.uid_ = knp::core::UID(boost::uuids::name_generator(core::UID(ns_uid))(cpu_name_.c_str()));
|
|
}
|
|
|
|
|
|
CPU::CPU(CPU&& other) : cpu_name_{std::move(other.cpu_name_)}, power_meter_{std::move(other.power_meter_)} {}
|
|
|
|
|
|
CPU::~CPU() {}
|
|
|
|
|
|
CPU& CPU::operator=(CPU&& other) noexcept
|
|
{
|
|
cpu_name_.swap(other.cpu_name_);
|
|
power_meter_.swap(other.power_meter_);
|
|
return *this;
|
|
}
|
|
|
|
|
|
knp::core::DeviceType CPU::get_type() const
|
|
{
|
|
return knp::core::DeviceType::CPU;
|
|
}
|
|
|
|
|
|
const std::string CPU::get_name() const
|
|
{
|
|
return cpu_name_;
|
|
}
|
|
|
|
|
|
uint32_t CPU::get_socket_number() const
|
|
{
|
|
return cpu_num_;
|
|
}
|
|
|
|
|
|
float CPU::get_power() const
|
|
{
|
|
return power_meter_->get_power();
|
|
}
|
|
|
|
|
|
KNP_DECLSPEC std::vector<CPU> list_processors()
|
|
{
|
|
auto* pcm_instance = pcm::PCM::getInstance();
|
|
if (nullptr == pcm_instance)
|
|
{
|
|
static constexpr const char* msg{"PCM instance received an error."};
|
|
SPDLOG_WARN(msg);
|
|
throw std::logic_error(msg);
|
|
}
|
|
|
|
try
|
|
{
|
|
const pcm::PCM::ErrorCode status = pcm_instance->program(pcm::PCM::DEFAULT_EVENTS, nullptr, true, ::getpid());
|
|
check_pcm_status(status);
|
|
}
|
|
catch (const std::exception& e)
|
|
{
|
|
SPDLOG_WARN("{}", e.what());
|
|
throw;
|
|
}
|
|
|
|
std::vector<CPU> result;
|
|
result.reserve(pcm_instance->getNumSockets());
|
|
|
|
for (uint32_t sock_num = 0; sock_num < pcm_instance->getNumSockets(); ++sock_num)
|
|
{
|
|
if (pcm_instance->isSocketOnline(sock_num)) result.push_back(CPU(sock_num));
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
} // namespace knp::devices::cpu
|