diff --git a/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc b/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc index 5556ee7c467..ea24a3faa70 100644 --- a/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc +++ b/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc @@ -143,6 +143,12 @@ SimpleOrcJIT::SimpleOrcJIT( main_jit_dylib_->addGenerator( std::make_unique(*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> SimpleOrcJIT::Create(