Eliminate per-op allocation of gpu device wrapper. The PerOpGpuDevice is allocated once in the OpKernelContext::Params struct, then re-used every time a new OpKernelContext uses the Params. Thus in the executor, as long as there is more work to do the PerOpGpuDevice is not freed.

Change: 112909215

1 files changed, 2 insertions, 2 deletions

diff --git a/tensorflow/core/kernels/restore_op_test.cc b/tensorflow/core/kernels/restore_op_test.cc
index 1dbd452843..9c5e52c574 100644
--- a/tensorflow/core/kernels/restore_op_test.cc
+++ b/tensorflow/core/kernels/restore_op_test.cc
@@ -168,7 +168,7 @@ TEST_F(RestoreOpTest, RestoreSimple) {
     checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper;
     params.slice_reader_cache = &slice_reader_cache_wrapper;
 
-    OpKernelContext ctx(params);
+    OpKernelContext ctx(&params);
     op->Compute(&ctx);
     EXPECT_OK(ctx.status());
   }
@@ -392,7 +392,7 @@ TEST_F(RestoreSliceOpTest, RestoreInt) {
     checkpoint::TensorSliceReaderCacheWrapper slice_reader_cache_wrapper;
     params.slice_reader_cache = &slice_reader_cache_wrapper;
 
-    OpKernelContext ctx(params);
+    OpKernelContext ctx(&params);
     op->Compute(&ctx);
     EXPECT_OK(ctx.status());
   }