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

[XLA:CPU] Fix OrcV2 on Windows

Browse Source 
PiperOrigin-RevId: 338285206
Change-Id: I50d14361be3e5a6a98bdc58032463f7b18f0e372
This commit is contained in:
Benjamin Kramer 2020-10-21 10:04:18 -07:00 committed by TensorFlower Gardener
parent 816a1177dc
commit fb6b46e7ad
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -143,6 +143,12 @@ SimpleOrcJIT::SimpleOrcJIT(
main_jit_dylib_->addGenerator(
std::make_unique<RuntimeSymbolGenerator>(*this));
object_layer_.registerJITEventListener(*this);
// Copied from LLJIT, required to find symbols on Windows.
if (target_machine_->getTargetTriple().isOSBinFormatCOFF()) {
object_layer_.setOverrideObjectFlagsWithResponsibilityFlags(true);
object_layer_.setAutoClaimResponsibilityForObjectSymbols(true);
}
}
llvm::Expected<std::unique_ptr<SimpleOrcJIT>> SimpleOrcJIT::Create(