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

Avoid unnecessary copying of map data during visitation

Browse Source 
PiperOrigin-RevId: 158141962
This commit is contained in:
A. Unique TensorFlower 2017-06-06 08:12:49 -07:00 committed by TensorFlower Gardener
parent 2e7e1d57bd
commit 9184726ed0
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/compiler/xla/service/llvm_ir/ir_array.cc
View File

@ -236,10 +236,8 @@ llvm::Value* IrArray::EmitReadArrayElement(const Index& index,
llvm::LoadInst* load = ir_builder->CreateLoad(element_address);
llvm_ir::SetTbaaForInstruction(load, GetShape(),
/*is_pointer_to=*/false);
for (const std::pair<int, llvm::MDNode*>& kind_md_pair : metadata_) {
int kind = kind_md_pair.first;
llvm::MDNode* md = kind_md_pair.second;
load->setMetadata(kind, md);
for (const auto& kind_md_pair : metadata_) {
load->setMetadata(kind_md_pair.first, kind_md_pair.second);
}
return load;
}
@ -250,11 +248,9 @@ void IrArray::EmitWriteArrayElement(const Index& index, llvm::Value* value,
llvm::StoreInst* store = ir_builder->CreateStore(value, element_address);
llvm_ir::SetTbaaForInstruction(store, GetShape(),
/*is_pointer_to=*/false);
for (const std::pair<int, llvm::MDNode*>& kind_md_pair : metadata_) {
int kind = kind_md_pair.first;
CHECK_NE(kind, llvm::LLVMContext::MD_invariant_load);
llvm::MDNode* md = kind_md_pair.second;
store->setMetadata(kind, md);
for (const auto& kind_md_pair : metadata_) {
CHECK_NE(kind_md_pair.first, llvm::LLVMContext::MD_invariant_load);
store->setMetadata(kind_md_pair.first, kind_md_pair.second);
}
}