intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-05 09:09:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add support for alarm in Windows ULT

Browse Source 
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2023-11-27 09:59:46 +00:00
committed by Compute-Runtime-Automation
parent c7daa7a960
commit 8e3caef814
7 changed files with 58 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
shared/test/common/os_interface/windows/signal_utils.cpp
View File

@@ -11,12 +11,21 @@
#include "gtest/gtest.h"
#include <condition_variable>
#include <io.h>
#include <signal.h>
#include <thread>
std::string lastTest("");
static int newStdOut = -1;
namespace NEO {
extern const char *executionName;
extern unsigned int ultIterationMaxTime;
} // namespace NEO
std::unique_ptr<std::thread> alarmThread;
LONG WINAPI ultExceptionFilter(
_In_ struct _EXCEPTION_POINTERS *exceptionInfo) {
std::cout << "UnhandledException: 0x" << std::hex << exceptionInfo->ExceptionRecord->ExceptionCode << std::dec
@@ -49,9 +58,48 @@ int setAbrt(bool enableAbrt) {
}
int setAlarm(bool enableAlarm) {
std::cout << "enable SIGALRM handler: " << enableAlarm << std::endl;
if (enableAlarm) {
std::string envVar = std::string("NEO_") + NEO::executionName + "_DISABLE_TEST_ALARM";
char *envValue = getenv(envVar.c_str());
if (envValue != nullptr) {
enableAlarm = false;
std::cout << "WARNING: SIGALRM handler disabled by environment variable: " << envVar << std::endl;
}
}
if (enableAlarm) {
std::condition_variable threadStarted;
alarmThread = std::make_unique<std::thread>([&]() {
auto currentUltIterationMaxTime = NEO::ultIterationMaxTime;
auto ultIterationMaxTimeEnv = getenv("NEO_ULT_ITERATION_MAX_TIME");
if (ultIterationMaxTimeEnv != nullptr) {
currentUltIterationMaxTime = atoi(ultIterationMaxTimeEnv);
}
unsigned int alarmTime = currentUltIterationMaxTime * ::testing::GTEST_FLAG(repeat);
std::cout << "set timeout to: " << alarmTime << std::endl;
threadStarted.notify_all();
std::this_thread::sleep_for(std::chrono::seconds(alarmTime));
printf("timeout on %s\n", lastTest.c_str());
abort();
});
std::mutex mtx;
std::unique_lock<std::mutex> lock(mtx);
threadStarted.wait(lock);
}
return 0;
}
int setSegv(bool enableSegv) {
return 0;
}
void cleanupSignals() {
if (alarmThread) {
alarmThread->detach();
alarmThread.reset();
}
}