compute-runtime/shared/test/unit_test/utilities/cpuintrinsics_tests.cpp

/*
 * Copyright (C) 2020-2022 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#include "shared/source/utilities/cpuintrinsics.h"

#include "gtest/gtest.h"

#include <atomic>
#include <cstdint>

namespace CpuIntrinsicsTests {
extern std::atomic<uintptr_t> lastClFlushedPtr;
extern std::atomic<uint32_t> pauseCounter;
extern std::atomic<uint32_t> sfenceCounter;
} // namespace CpuIntrinsicsTests

TEST(CpuIntrinsicsTest, whenClFlushIsCalledThenExpectToPassPtrToSystemCall) {
    uintptr_t flushAddr = 0x1234;
    void const *ptr = reinterpret_cast<void const *>(flushAddr);
    NEO::CpuIntrinsics::clFlush(ptr);
    EXPECT_EQ(flushAddr, CpuIntrinsicsTests::lastClFlushedPtr);
}

TEST(CpuIntrinsicsTest, whenPauseCalledThenExpectToIncreaseCounter) {
    uint32_t oldCount = CpuIntrinsicsTests::pauseCounter.load();
    NEO::CpuIntrinsics::pause();
    EXPECT_EQ(oldCount + 1, CpuIntrinsicsTests::pauseCounter);
}

TEST(CpuIntrinsicsTest, whenSfenceCalledThenExpectToIncreaseCounter) {
    uint32_t oldCount = CpuIntrinsicsTests::sfenceCounter.load();
    NEO::CpuIntrinsics::sfence();
    EXPECT_EQ(oldCount + 1, CpuIntrinsicsTests::sfenceCounter);
}