}

  absl::StatusOr<std::unique_ptr<xla::PjRtLoadedExecutable>>
  BuildSamplePjRtExecutable() {
    return pjrt_compiler_client_->BuildExecutable(DefaultPjRtOptions(),
                                                  compilation_result_add_);
  }

  absl::StatusOr<XlaCompiler::CompilationResult> BuildSampleCompilationResult(
      bool mul = false) {