aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/user_computation.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/service/user_computation.cc')
-rw-r--r--tensorflow/compiler/xla/service/user_computation.cc6
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/compiler/xla/service/user_computation.cc b/tensorflow/compiler/xla/service/user_computation.cc
index 8d5bb08e51..e9d182509b 100644
--- a/tensorflow/compiler/xla/service/user_computation.cc
+++ b/tensorflow/compiler/xla/service/user_computation.cc
@@ -2927,9 +2927,8 @@ void ComputationLowerer::Visit(
case OpRequest::kRecvRequest: {
const RecvRequest& recv_request = request.request().recv_request();
- HloInstruction* recv = add_instruction(HloInstruction::CreateRecv(
+ hlo_instruction = add_instruction(HloInstruction::CreateRecv(
request.output_shape(), recv_request.channel_handle().handle()));
- hlo_instruction = add_instruction(HloInstruction::CreateRecvDone(recv));
break;
}
@@ -3121,9 +3120,8 @@ void ComputationLowerer::Visit(
case OpRequest::kSendRequest: {
const SendRequest& send_request = request.request().send_request();
HloInstruction* operand = lookup_instruction(send_request.operand());
- HloInstruction* send = add_instruction(HloInstruction::CreateSend(
+ hlo_instruction = add_instruction(HloInstruction::CreateSend(
operand, send_request.channel_handle().handle()));
- hlo_instruction = add_instruction(HloInstruction::CreateSendDone(send));
break;
}