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

[XLA:CPU] When built with MSAN, unpoison outfed buffers to suppress false-positive MSAN warnings.

Browse Source 
PiperOrigin-RevId: 354080879
Change-Id: If5b257d2a0fd992b79806054b43eef4b77f66d72
This commit is contained in:
Peter Hawkins 2021-01-27 06:40:40 -08:00 committed by TensorFlower Gardener
parent f898ddfbf5
commit 7d5328c5df
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/compiler/xla/service/cpu/ir_emitter.cc
View File

@ -487,6 +487,17 @@ Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
MemCpy(acquired_pointer, /*DstAlign=*/llvm::Align(1),
program_buffer_address,
/*SrcAlign=*/llvm::Align(1), length_32);
if (emit_code_for_msan_) {
// Mark the outfed data as initialized for msan. The buffer gets read by
// the host code, which might be msan-instrumented.
// TODO(b/66051036): Run the msan instrumentation pass instead.
const llvm::DataLayout& dl = module_->getDataLayout();
llvm::Type* intptr_type = b_.getIntPtrTy(dl);
EmitCallToFunc(
"__msan_unpoison",
{acquired_pointer, llvm::ConstantInt::get(intptr_type, length)},
b_.getVoidTy());
}
}
const char* release_func_name =