pghafari
c221406681
making sure the current thread is not null before calling ihipDestroyDevice Change-Id: Ib75e3f9ee1c4e0065986e3fed1065907220f4379
66 строки
2.1 KiB
C++
66 строки
2.1 KiB
C++
/* Copyright (c) 2008 - 2022 Advanced Micro Devices, Inc.
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE. */
|
|
|
|
#include "thread/thread.hpp"
|
|
|
|
#include <windows.h>
|
|
#include <iostream>
|
|
|
|
void ihipDestroyDevice();
|
|
|
|
#ifdef DEBUG
|
|
static int reportHook(int reportType, char* message, int* returnValue) {
|
|
if (returnValue) {
|
|
*returnValue = 1;
|
|
}
|
|
std::cerr << message;
|
|
::exit(3);
|
|
return TRUE;
|
|
}
|
|
#endif // DEBUG
|
|
|
|
extern "C" BOOL WINAPI DllMain(HINSTANCE hinst, DWORD reason, LPVOID reserved) {
|
|
switch (reason) {
|
|
case DLL_PROCESS_ATTACH:
|
|
#ifdef DEBUG
|
|
if (!::getenv("AMD_OCL_ENABLE_MESSAGE_BOX")) {
|
|
_CrtSetReportHook(reportHook);
|
|
_set_error_mode(_OUT_TO_STDERR);
|
|
}
|
|
#endif // DEBUG
|
|
break;
|
|
case DLL_PROCESS_DETACH: {
|
|
amd::Thread* thread = amd::Thread::current();
|
|
if (!(thread != nullptr ||
|
|
((thread = new amd::HostThread()) != nullptr && thread == amd::Thread::current()))) {
|
|
return true;
|
|
}
|
|
ihipDestroyDevice();
|
|
} break;
|
|
case DLL_THREAD_DETACH: {
|
|
amd::Thread* thread = amd::Thread::current();
|
|
delete thread;
|
|
} break;
|
|
default:
|
|
break;
|
|
}
|
|
return true;
|
|
}
|