diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/llvm_ir/ir_array.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/llvm_ir/ir_array.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/llvm_ir/ir_array.h b/tensorflow/compiler/xla/service/llvm_ir/ir_array.h index 4648c6d7ac..fb1f01ab6b 100644 --- a/tensorflow/compiler/xla/service/llvm_ir/ir_array.h +++ b/tensorflow/compiler/xla/service/llvm_ir/ir_array.h @@ -248,6 +248,11 @@ class IrArray { void AnnotateLoadStoreInstructionWithMetadata( llvm::Instruction* instruction) const; + // Uses the metadata for a parameter IrArray to annotate the load/store of the + // tile buffer for the parameter. + void AnnotateBufferLoadStoreInstructionWithMetadata( + llvm::Instruction* instruction) const; + // Emit IR to read an array element at the given index. Returns the read // result (effectively, a Value loaded from memory). This method seamlessly // handles scalar shapes by broadcasting their value to all indices (index is |