Internal change

PiperOrigin-RevId: 201645254

1 files changed, 2 insertions, 8 deletions

diff --git a/tensorflow/compiler/jit/xla_compile_on_demand_op.cc b/tensorflow/compiler/jit/xla_compile_on_demand_op.cc
index 9beeb3517e..26f350855d 100644
--- a/tensorflow/compiler/jit/xla_compile_on_demand_op.cc
+++ b/tensorflow/compiler/jit/xla_compile_on_demand_op.cc
@@ -71,14 +71,8 @@ Status XlaCompileOnDemandOp::Run(OpKernelContext* ctx,
   run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device());
   run_options.set_rng_seed(ctx->step_id());
 
-  xla::StatusOr<xla::ScopedShapedBuffer> run_result;
-  {
-    // TODO(b/110383871): fix concurrency problems and remove this mutex.
-    static mutex* mu = new mutex;
-    mutex_lock lock(*mu);
-
-    run_result = executable->Run(launch_context.arguments(), run_options);
-  }
+  xla::StatusOr<xla::ScopedShapedBuffer> run_result =
+      executable->Run(launch_context.arguments(), run_options);
   TF_RETURN_IF_ERROR(run_result.status());
 
   launch_context.PopulateOutputs(ctx, result, run_result.ConsumeValueOrDie());