Add default implementation for Finalize().

tensorflower-gardener · tensorflower-gardener · commit 36f863b294fa · 2024-09-10T15:36:06.000-07:00
PiperOrigin-RevId: 673107887
diff --git a/tensorflow/core/tfrt/tfrt_session/tfrt_session.cc b/tensorflow/core/tfrt/tfrt_session/tfrt_session.cc
@@ -481,6 +481,8 @@ class TfrtSession : public tensorflow::Session {
     return absl::OkStatus();
   }
 
+  Status Finalize() override { return absl::OkStatus(); }
+
  private:
   tfrt::HostContext* GetHostContext() {
     return runtime_->core_runtime()->GetHostContext();
diff --git a/tensorflow/core/tfrt/tfrt_session/tfrt_session_test.cc b/tensorflow/core/tfrt/tfrt_session/tfrt_session_test.cc
@@ -399,6 +399,8 @@ TEST_F(TfrtSessionTest, Callable) {
   TF_ASSERT_OK(session_->ReleaseCallable(callable_handle));
 }
 
+TEST_F(TfrtSessionTest, Finalize) { TF_ASSERT_OK(session_->Finalize()); }
+
 TEST_F(TfrtSessionTest, WithTargetNodes) {
   std::vector<Tensor> outputs;
   TF_ASSERT_OK(session_->Run(inputs_, output_tensor_names_, target_node_names_,

Original file line number	Diff line number	Diff line change
`@@ -481,6 +481,8 @@ class TfrtSession : public tensorflow::Session {`
`481`	`481`	`return absl::OkStatus();`
`482`	`482`	`}`
`483`	`483`
	`484`	`+ Status Finalize() override { return absl::OkStatus(); }`
	`485`	`+`
`484`	`486`	`private:`
`485`	`487`	`tfrt::HostContext* GetHostContext() {`
`486`	`488`	`return runtime_->core_runtime()->GetHostContext();`
Original file line number	Diff line number	Diff line change
`@@ -399,6 +399,8 @@ TEST_F(TfrtSessionTest, Callable) {`
`399`	`399`	`TF_ASSERT_OK(session_->ReleaseCallable(callable_handle));`
`400`	`400`	`}`
`401`	`401`
	`402`	`+TEST_F(TfrtSessionTest, Finalize) { TF_ASSERT_OK(session_->Finalize()); }`
	`403`	`+`
`402`	`404`	`TEST_F(TfrtSessionTest, WithTargetNodes) {`
`403`	`405`	`std::vector<Tensor> outputs;`
`404`	`406`	`TF_ASSERT_OK(session_->Run(inputs_, output_tensor_names_, target_node_names_,`