--- /dev/null
+# RUN: llvm-exegesis -mode=uops -opcode-name=FLDENVm,FLDL2E -repetition-mode=duplicate | FileCheck %s
+
+CHECK: mode: uops
+CHECK-NEXT: key:
+CHECK-NEXT: instructions:
+CHECK-NEXT: FLDENVm
SmallVector<StringRef, 2> CounterNames;
StringRef(Counters).split(CounterNames, '+');
char *const ScratchPtr = Scratch->ptr();
+ const ExegesisTarget &ET = State.getExegesisTarget();
for (auto &CounterName : CounterNames) {
CounterName = CounterName.trim();
- auto CounterOrError =
- State.getExegesisTarget().createCounter(CounterName, State);
+ auto CounterOrError = ET.createCounter(CounterName, State);
if (!CounterOrError)
return CounterOrError.takeError();
.concat(std::to_string(Reserved)));
Scratch->clear();
{
+ auto PS = ET.withSavedState();
CrashRecoveryContext CRC;
CrashRecoveryContext::Enable();
const bool Crashed = !CRC.RunSafely([this, Counter, ScratchPtr]() {
Counter->stop();
});
CrashRecoveryContext::Disable();
+ PS.reset();
if (Crashed) {
std::string Msg = "snippet crashed while running";
#ifdef LLVM_ON_UNIX
return *Found->PCI;
}
+ExegesisTarget::SavedState::~SavedState() {} // anchor.
+
namespace {
// Default implementation.
// counters are defined for this CPU).
const PfmCountersInfo &getPfmCounters(StringRef CpuName) const;
+ // Saves the CPU state that needs to be preserved when running a benchmark,
+ // and returns and RAII object that restores the state on destruction.
+ // By default no state is preserved.
+ struct SavedState {
+ virtual ~SavedState();
+ };
+ virtual std::unique_ptr<SavedState> withSavedState() const {
+ return std::make_unique<SavedState>();
+ }
+
private:
virtual bool matchesArch(Triple::ArchType Arch) const = 0;
#include <memory>
#include <string>
#include <vector>
+#if defined(_MSC_VER)
+#include <immintrin.h>
+#endif
namespace llvm {
namespace exegesis {
namespace {
+class X86SavedState : public ExegesisTarget::SavedState {
+public:
+ X86SavedState() {
+#if defined(_MSC_VER)
+ _fxsave64(FPState);
+#elif defined(__GNUC__)
+ __builtin_ia32_fxsave64(FPState);
+#endif
+ }
+
+ ~X86SavedState() {
+ // Restoring the X87 state does not flush pending exceptions, make sure
+ // these exceptions are flushed now.
+#if defined(_MSC_VER)
+ _clearfp();
+ _fxrstor64(FPState);
+#elif defined(__GNUC__)
+ asm volatile("fwait");
+ __builtin_ia32_fxrstor64(FPState);
+#endif
+ }
+
+private:
+ alignas(16) char FPState[512];
+};
+
class ExegesisX86Target : public ExegesisTarget {
public:
ExegesisX86Target() : ExegesisTarget(X86CpuPfmCounters) {}
#endif
}
+ std::unique_ptr<SavedState> withSavedState() const override {
+ return std::make_unique<X86SavedState>();
+ }
+
static const unsigned kUnavailableRegisters[4];
};