experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add __chkstk to the list of host symbols known to XLA/CPU on windows.

Browse Source 
PiperOrigin-RevId: 294337304
Change-Id: I61f0a386d2a2dbd7cd68c105d3081448bc639ddf
This commit is contained in:
Artem Belevich 2020-02-10 17:01:07 -08:00 committed by TensorFlower Gardener
parent 432ef2bee2
commit 12b968f470
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc
View File

@ -212,6 +212,12 @@ llvm::JITSymbol SimpleOrcJIT::FindCompiledSymbol(const std::string& name) {
return nullptr; return nullptr;
} }
#if defined(PLATFORM_WINDOWS)
// This function is used by compiler-generated code on windows, but it's not
// declared anywhere. The signature does not matter, we just need the address.
extern "C" void __chkstk(size_t);
#endif
namespace { namespace {
// Register some known symbols with the CustomCallTargetRegistry. // Register some known symbols with the CustomCallTargetRegistry.
bool RegisterKnownJITSymbols() { bool RegisterKnownJITSymbols() {
@ -364,6 +370,10 @@ bool RegisterKnownJITSymbols() {
reinterpret_cast<void*>(__msan_unpoison), "Host"); reinterpret_cast<void*>(__msan_unpoison), "Host");
#endif #endif
#if defined(PLATFORM_WINDOWS)
registry->Register("__chkstk", reinterpret_cast<void*>(__chkstk), "Host");
#endif
return true; return true;
} }