OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Remove unnecessary calls of WorkQueueInterface::InitializeRequest() in TFRT GPU

Browse Source 
PiperOrigin-RevId: 403176042
Change-Id: Id8c093033c11534400d0fb40564e7c442465d64b
This commit is contained in:
Kuangyuan Chen 2021-10-14 13:41:21 -07:00 committed by TensorFlower Gardener
parent ecc1fc7be3
commit d1d9afda4d
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/service/gpu/bef_thunk.cc
View File

@ -487,10 +487,6 @@ static StatusOr<std::unique_ptr<tfrt::ExecutionContext>> CreateExecutionContext(
TF_ASSIGN_OR_RETURN(auto runtime_and_queue, GetCoreRuntimeAndWorkQueue());
tfrt::RequestContextBuilder request_context_builder(
runtime_and_queue.core_runtime->GetHostContext(), resource_context);
tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr;
TF_RETURN_IF_ERROR(runtime_and_queue.work_queue->InitializeRequest(
&request_context_builder, &intra_op_threadpool));
TF_RETURN_IF_ERROR(build_request_context(request_context_builder));
auto expected_req_ctx = std::move(request_context_builder).build();