99c0a73f91
d0488d932 oaknut: 2.0.0 40ad78bbf oaknut: Implement DualCodeBlock and related support 9f131cfb5 oaknut: add configuration for standalone installation 69799b43c oaknut: Test building for Android on CI 1d51f5512 oaknut: 1.2.2 918bd94f0 oaknut: Eliminate -Wconversion warnings 316d8869e oaknut: Fix edgecases in MOVP2R on +/-4GiB boundary d8634eaa1 oaknut: Fix page boundary error in ADP d0ca9a24e oaknut: Update README examples for CPU feature detection dbeec268b oaknut: feature_detection_freebsd: Warn about incompatibility with earlier FreeBSD versions 86e5386e2 oaknut: feature_detect: Support NetBSD df4cf2d48 oaknut: feature_detect: Support OpenBSD 99dfff25a oaknut: feature_detection: Read ID registers 319b3d2c9 oaknut: Add basic CPU feature detection 23e9ddb4c oaknut: CI: Don't run slow tests on OpenBSD 734f1bdb4 oaknut: CI: Use up-to-date qemu f462c9774 oaknut: CI: Build on OpenBSD 19cd42204 oaknut: code_block: Add NetBSD and OpenBSD support 18b86a3ec oaknut: SystemReg: Add more EL0 accessible registers 53c43bf0c oaknut/tests: Reduce iterations for MOVP2R cc37df19e oaknut: Test on FreeBSD a66b32d26 oaknut: Fix crossing sign boundary in PageOffset 206468d72 oaknut: CI: Add macos-arm64 build e6eecc3f9 oaknut: 1.2.1 4252d8f4a oaknut: CMakeLists: Warnings are errors on MSVC 408eed65f oaknut: arm64_encode_helpers: remove unreachable code bfc8eedfb oaknut: arm64_encode_helpers: p maybe unused ff4456eca oaknut: Avoid negation of unsigned values b4ac8fd6c oaknut: Fix MOV for applications of MOVN 0575cadc4 oaknut: Disable certain functionality where absolute addressing is not available 394a3c8f0 oaknut: Appease MSVC 011183670 oaknut: 1.2.0 e83c9f327 oaknut: Add VectorCodeGenerator 5eb122cc5 oaknut: Tidy up public header 45c5a7b25 oaknut: Fix clang-format errors 36243256f oaknut: Add `const` qualifier to `AddrOffset` ctor 4af500cb5 oaknut: Add `ptr` accessor to `Label` bccb06669 oaknut: CodeGenerator const correctness da0590a86 oaknut: github: Update package repositories git-subtree-dir: externals/oaknut git-subtree-split: d0488d9320ae673167dd9117223e3453d5ff102f
99 lines
3.6 KiB
C++
99 lines
3.6 KiB
C++
// SPDX-FileCopyrightText: Copyright (c) 2023 merryhime <https://mary.rs>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
#ifndef WIN32_LEAN_AND_MEAN
|
|
# define WIN32_LEAN_AND_MEAN
|
|
#endif
|
|
|
|
#include <windows.h>
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <optional>
|
|
|
|
#include <processthreadsapi.h>
|
|
|
|
#include "oaknut/feature_detection/cpu_feature.hpp"
|
|
#include "oaknut/feature_detection/id_registers.hpp"
|
|
|
|
namespace oaknut {
|
|
|
|
namespace detail {
|
|
|
|
inline std::optional<std::uint64_t> read_registry_hklm(const std::string& subkey, const std::string& name)
|
|
{
|
|
std::uint64_t value;
|
|
DWORD value_len = sizeof(value);
|
|
if (::RegGetValueA(HKEY_LOCAL_MACHINE, subkey.c_str(), name.c_str(), RRF_RT_REG_QWORD, nullptr, &value, &value_len) == ERROR_SUCCESS) {
|
|
return value;
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
inline std::uint64_t read_id_register(std::size_t core_index, const std::string& name)
|
|
{
|
|
return read_registry_hklm("HARDWARE\\DESCRIPTION\\System\\CentralProcessor\\" + std::to_string(core_index), "CP " + name).value_or(0);
|
|
}
|
|
|
|
} // namespace detail
|
|
|
|
// Ref: https://learn.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-isprocessorfeaturepresent
|
|
|
|
inline CpuFeatures detect_features_via_IsProcessorFeaturePresent()
|
|
{
|
|
CpuFeatures result;
|
|
|
|
if (::IsProcessorFeaturePresent(30)) // PF_ARM_V8_CRYPTO_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::AES, CpuFeature::PMULL, CpuFeature::SHA1, CpuFeature::SHA256};
|
|
if (::IsProcessorFeaturePresent(31)) // PF_ARM_V8_CRC32_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::CRC32};
|
|
if (::IsProcessorFeaturePresent(34)) // PF_ARM_V81_ATOMIC_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::LSE};
|
|
if (::IsProcessorFeaturePresent(43)) // PF_ARM_V82_DP_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::DotProd};
|
|
if (::IsProcessorFeaturePresent(44)) // PF_ARM_V83_JSCVT_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::JSCVT};
|
|
if (::IsProcessorFeaturePresent(45)) // PF_ARM_V83_LRCPC_INSTRUCTIONS_AVAILABLE
|
|
result |= CpuFeatures{CpuFeature::LRCPC};
|
|
|
|
return result;
|
|
}
|
|
|
|
inline CpuFeatures detect_features()
|
|
{
|
|
CpuFeatures result{CpuFeature::FP, CpuFeature::ASIMD};
|
|
result |= detect_features_via_IsProcessorFeaturePresent();
|
|
return result;
|
|
}
|
|
|
|
inline std::size_t get_core_count()
|
|
{
|
|
::SYSTEM_INFO sys_info;
|
|
::GetSystemInfo(&sys_info);
|
|
return sys_info.dwNumberOfProcessors;
|
|
}
|
|
|
|
inline std::optional<id::IdRegisters> read_id_registers(std::size_t core_index)
|
|
{
|
|
return id::IdRegisters{
|
|
detail::read_id_register(core_index, "4000"),
|
|
id::Pfr0Register{detail::read_id_register(core_index, "4020")},
|
|
id::Pfr1Register{detail::read_id_register(core_index, "4021")},
|
|
id::Pfr2Register{detail::read_id_register(core_index, "4022")},
|
|
id::Zfr0Register{detail::read_id_register(core_index, "4024")},
|
|
id::Smfr0Register{detail::read_id_register(core_index, "4025")},
|
|
id::Isar0Register{detail::read_id_register(core_index, "4030")},
|
|
id::Isar1Register{detail::read_id_register(core_index, "4031")},
|
|
id::Isar2Register{detail::read_id_register(core_index, "4032")},
|
|
id::Isar3Register{detail::read_id_register(core_index, "4033")},
|
|
id::Mmfr0Register{detail::read_id_register(core_index, "4038")},
|
|
id::Mmfr1Register{detail::read_id_register(core_index, "4039")},
|
|
id::Mmfr2Register{detail::read_id_register(core_index, "403A")},
|
|
id::Mmfr3Register{detail::read_id_register(core_index, "403B")},
|
|
id::Mmfr4Register{detail::read_id_register(core_index, "403C")},
|
|
};
|
|
}
|
|
|
|
} // namespace oaknut
|