blob: 331f9a7d7f46b0977786c66e7b6cb44253b68360 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
/* ========================================
Copyright (C) 2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
======================================== */
#pragma once
#include <misc/BenchKit/Chronometer.h>
namespace Kernel {
struct HWChronoTraits;
template <typename ChronoTraits = HWChronoTraits>
class HWChrono;
/// @brief BenchKit chrono logic for x64/ARM64.
struct HWChronoTraits final {
private:
STATIC UInt64 TickImpl_(void) {
#ifdef __NE_AMD64__
UInt64 a = 0, d = 0;
asm volatile("rdtsc" : "=a"(a), "=d"(d));
return (d << 32) | a;
#elif defined(__NE_ARM64__)
UInt64 result;
asm volatile("mrs %0, cntvct_el1" : "=r"(result));
return result;
#else
#error !!! no backend defined !!!
#endif
}
friend HWChrono<HWChronoTraits>;
};
/// @brief hardware chronometer implementation using a trait to extract the data.
template <typename ChronoTraits>
class HWChrono BENCHKIT_INTERFACE {
public:
HWChrono() = default;
virtual ~HWChrono() override = default;
NE_COPY_DEFAULT(HWChrono)
public:
Void Start() override { fStart = ChronoTraits::TickImpl_(); }
Void Stop() override { fStop = ChronoTraits::TickImpl_(); }
Void Reset() override {
fStart = 0;
fStop = 0;
}
UInt64 GetElapsedTime() const override { return fStop - fStart; }
private:
UInt64 fStart{};
UInt64 fStop{};
};
} // namespace Kernel
|
