Skip to content

Commit 6516fe8

Browse files
changhuilinGoogle-ML-Automation
authored andcommitted
Change PjRtClient::Compile() to return an unloaded executable.
PiperOrigin-RevId: 738932786
1 parent 22b71a5 commit 6516fe8

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

xla/pjrt/pjrt_client.h

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -594,25 +594,24 @@ class PjRtClient {
594594
}
595595

596596
// Compile `computation` with given `options`.
597-
// ABSL_DEPRECATED("Use CompileAndLoad() below instead")
598-
virtual absl::StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(
597+
virtual absl::StatusOr<std::unique_ptr<PjRtExecutable>> Compile(
599598
const XlaComputation& computation, CompileOptions options) {
600-
return CompileAndLoad(computation, options);
599+
return Unimplemented("Compile with XlaComputation is not supported.");
601600
}
602601
virtual absl::StatusOr<std::unique_ptr<PjRtLoadedExecutable>> CompileAndLoad(
603602
const XlaComputation& computation, CompileOptions options) {
604-
return Unimplemented("Compile with XlaComputation is not supported.");
603+
return Unimplemented(
604+
"CompileAndLoad with XlaComputation is not supported.");
605605
}
606606

607607
// Variant of `Compile` that accepts an MLIR module.
608-
// ABSL_DEPRECATED("Use CompileAndLoad() below instead")
609-
virtual absl::StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(
608+
virtual absl::StatusOr<std::unique_ptr<PjRtExecutable>> Compile(
610609
mlir::ModuleOp module, CompileOptions options) {
611-
return CompileAndLoad(module, options);
610+
return Unimplemented("Compile with MLIR Module is not supported.");
612611
}
613612
virtual absl::StatusOr<std::unique_ptr<PjRtLoadedExecutable>> CompileAndLoad(
614613
mlir::ModuleOp module, CompileOptions options) {
615-
return Unimplemented("Compile with MLIR Module is not supported.");
614+
return Unimplemented("CompileAndLoad with MLIR Module is not supported.");
616615
}
617616

618617
// Deserializes a serialized executable as produced by

0 commit comments

Comments
 (0)