From 28db3a7eae4986e3e662de16188cf7a03be33768 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 20 Mar 2018 02:26:31 -0700 Subject: [PATCH] Fix bug PiperOrigin-RevId: 189712233 --- tensorflow/compiler/jit/xla_launch_util.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/compiler/jit/xla_launch_util.cc b/tensorflow/compiler/jit/xla_launch_util.cc index 076cbd2..bb7316c 100644 --- a/tensorflow/compiler/jit/xla_launch_util.cc +++ b/tensorflow/compiler/jit/xla_launch_util.cc @@ -169,7 +169,7 @@ void XlaComputationLaunchContext::PopulateOutputs( int output_num = 0; for (int i = 0; i < ctx->num_outputs(); ++i) { AllocatorAttributes alloc_attrs = ctx->output_alloc_attr(i); - Allocator* allocator = ctx->device()->GetAllocator(alloc_attrs); + Allocator* allocator = ctx->device()->GetAllocator({}); if (tensor_info_manager_ && !alloc_attrs.on_host()) { allocator = tensor_info_manager_; } -- 2.7.4