aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc')
-rw-r--r--tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc9
1 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc b/tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc
index 59ae5acd8b..8560e4296a 100644
--- a/tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc
+++ b/tensorflow/compiler/xla/service/cpu/parallel_loop_emitter.cc
@@ -25,8 +25,8 @@ namespace cpu {
ParallelLoopEmitter::ParallelLoopEmitter(
const llvm_ir::ElementGenerator& target_element_generator,
const llvm_ir::IrArray& target_array,
- const DynamicLoopBounds* dynamic_loop_bounds, llvm::IRBuilder<>* ir_builder)
- : LoopEmitter(target_element_generator, target_array, ir_builder),
+ const DynamicLoopBounds* dynamic_loop_bounds, llvm::IRBuilder<>* b)
+ : LoopEmitter(target_element_generator, target_array, b),
dynamic_loop_bounds_(dynamic_loop_bounds) {}
std::vector<llvm_ir::IrArray::Index>
@@ -37,7 +37,7 @@ ParallelLoopEmitter::EmitIndexAndSetExitBasicBlock(
CHECK(!ShapeUtil::IsTuple(shape_));
CHECK(!ShapeUtil::IsScalar(shape_));
- llvm_ir::ForLoopNest loop_nest(loop_name, ir_builder_);
+ llvm_ir::ForLoopNest loop_nest(loop_name, b_);
const int64 num_dims = shape_.dimensions_size();
llvm_ir::IrArray::Index array_index(index_type, num_dims);
@@ -65,8 +65,7 @@ ParallelLoopEmitter::EmitIndexAndSetExitBasicBlock(
}
}
// Point IR builder at inner loop BB.
- llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(),
- ir_builder_);
+ llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_);
// Set exit_bb_ to the exit block of the loop nest.
exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock();