experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Removing the unused HostComputeMetadata field.

Browse Source 
PiperOrigin-RevId: 337022622
Change-Id: Ibd191b66095b7df84173f42a4f115bab316ef625
This commit is contained in:
Henry Tan 2020-10-13 22:25:01 -07:00 committed by TensorFlower Gardener
parent c0d20ffd82
commit 016bcfcb19
4 changed files with 0 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/core/tpu/kernels/tpu_compilation_cache_rpc_support.cc
View File

@ -111,23 +111,6 @@ xla::StatusOr<std::vector<::grpc::Slice>> SerializeCacheEntryToBufferSlices(
}
header.set_is_empty(false);
HostComputeMetadataSerializedProto host_compute_metadata;
auto cleanup_host_compute_metadata =
xla::MakeCleanup([&host_compute_metadata]() {
if (host_compute_metadata.size > 0) {
stream_executor::tpu::SerializedProto_Free(host_compute_metadata);
}
});
Status get_host_compute_metadata_status =
tpu_program_group->SerializeHostComputeMetadata(cache_entry.core_index(),
&host_compute_metadata);
if (!get_host_compute_metadata_status.ok()) {
return errors::Internal("Failed to serialize host compute metadata.");
}
if (!header.mutable_host_compute_metadata()->ParseFromArray(
host_compute_metadata.bytes, host_compute_metadata.size)) {
return errors::Internal("Failed to deserialize host compute metadata.");
}
bool may_modify_variables =
tpu_program_group->may_modify_variables(cache_entry.core_index());

6
tensorflow/core/tpu/kernels/tpu_program_c_api.h
View File

@ -105,11 +105,6 @@ TFTPU_CAPI_EXPORT void TpuProgram_SerializeCompilerMetadata(
const XLA_TpuProgram* tpu_program,
CompilerMetadataSerializedProto* compiler_metadata, SE_Status* status);
// Gets host transfer metadata proto from a `tpu_program`.
TFTPU_CAPI_EXPORT void TpuProgram_SerializeHostComputeMetadata(
const XLA_TpuProgram* tpu_program,
HostComputeMetadataSerializedProto* host_compute_metadata,
SE_Status* status);
// Deserializes the `GetTpuProgramResponse` proto into an `XLA_TpuProgram`.
TFTPU_CAPI_EXPORT void TpuProgram_DeserializeFromGetTpuProgramResponseProto(
@ -132,7 +127,6 @@ struct TfTpu_TpuProgramApiFn {
TFTPU_ADD_FN_IN_STRUCT(TpuProgram_GetTpuProgram);
TFTPU_ADD_FN_IN_STRUCT(TpuProgram_SerializeTpuExecutable);
TFTPU_ADD_FN_IN_STRUCT(TpuProgram_SerializeCompilerMetadata);
TFTPU_ADD_FN_IN_STRUCT(TpuProgram_SerializeHostComputeMetadata);
TFTPU_ADD_FN_IN_STRUCT(TpuProgram_DeserializeFromGetTpuProgramResponseProto);
};

11
tensorflow/core/tpu/kernels/tpu_program_group.cc
View File

@ -333,16 +333,5 @@ Status TpuProgramGroup::SerializeCompilerMetadata(
tpu_programs_[index], compiler_metadata, status.c_status);
return status.status();
}
Status TpuProgramGroup::SerializeHostComputeMetadata(
int index,
HostComputeMetadataSerializedProto* host_compute_metadata) const {
CHECK_GE(index, 0);
CHECK_LT(index, tpu_programs_.size());
StatusHelper status;
TpuProgramApiFn()->TpuProgram_SerializeHostComputeMetadataFn(
tpu_programs_[index], host_compute_metadata, status.c_status);
return status.status();
}
} // namespace tpu
} // namespace tensorflow

1
tensorflow/core/tpu/tpu_library_init_fns.inc
View File

@ -77,7 +77,6 @@ tensorflow::Status SetTpuProgramStructFn(void* library_handle) {
TFTPU_SET_FN(tpu_program_fn, TpuProgram_GetTpuProgram);
TFTPU_SET_FN(tpu_program_fn, TpuProgram_SerializeTpuExecutable);
TFTPU_SET_FN(tpu_program_fn, TpuProgram_SerializeCompilerMetadata);
TFTPU_SET_FN(tpu_program_fn, TpuProgram_SerializeHostComputeMetadata);
TFTPU_SET_FN(tpu_program_fn,
TpuProgram_DeserializeFromGetTpuProgramResponseProto);