aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h')
-rw-r--r--tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h9
1 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h b/tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h
index a86e6e78c6..eee40b0e91 100644
--- a/tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h
+++ b/tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h
@@ -36,14 +36,13 @@ class HloToIrBindings {
public:
HloToIrBindings(const HloModule& module,
const BufferAssignment* buffer_assignment,
- llvm::IRBuilder<>* ir_builder, llvm::Module* llvm_module,
+ llvm::IRBuilder<>* b, llvm::Module* llvm_module,
bool is_nested)
: buffer_assignment_(buffer_assignment),
is_nested_(is_nested),
- ir_builder_(ir_builder),
+ b_(b),
module_(llvm_module),
- alias_analysis_(module, *buffer_assignment_,
- &ir_builder_->getContext()) {}
+ alias_analysis_(module, *buffer_assignment_, &b_->getContext()) {}
void EmitBasePointersForHlos(
tensorflow::gtl::ArraySlice<const HloInstruction*> io_hlos,
@@ -104,7 +103,7 @@ class HloToIrBindings {
const bool is_nested_;
- llvm::IRBuilder<>* ir_builder_;
+ llvm::IRBuilder<>* b_;
llvm::Module* module_;
// Stores the underlying llvm::IrArray for each HloInstruction.