pjrt_c_api_helpers.cc
imports xla/tsl/platform/errors.himports xla/tsl/platform/logging.h
SHA-256682f61725ab44ecb405a739617d3da4a163502eedb12c36221029af7d2101b51
MaleculeH(Db)Md(In₄)
Evidence
1/* Copyright 2022 The OpenXLA Authors.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16#include "xla/pjrt/c/pjrt_c_api_helpers.h"
17
18#include <algorithm>
19#include <cstddef>
20#include <cstdint>
21#include <functional>
22#include <memory>
23#include <string>
24#include <utility>
25#include <variant>
26#include <vector>
27
28#include "absl/container/flat_hash_map.h"
29#include "absl/container/inlined_vector.h"
30#include "absl/functional/any_invocable.h"
31#include "absl/log/check.h"
32#include "absl/log/log.h"
33#include "absl/status/status.h"
34#include "absl/status/statusor.h"
35#include "absl/strings/cord.h"
36#include "absl/strings/str_cat.h"
37#include "absl/strings/string_view.h"
38#include "absl/time/time.h"
39#include "absl/types/span.h"
40#include "stablehlo/dialect/Version.h"
41#include "xla/future.h"
42#include "xla/layout.h"
43#include "xla/pjrt/c/pjrt_c_api.h"
44#include "xla/pjrt/c/pjrt_c_api_layouts_extension.h"
45#include "xla/pjrt/c/pjrt_c_api_memory_descriptions_extension.h"
46#include "xla/pjrt/c/pjrt_c_api_profiler_extension.h"
47#include "xla/pjrt/compiled_memory_stats.h"
48#include "xla/pjrt/distributed/key_value_store_interface.h"
49#include "xla/pjrt/pjrt_client.h"
50#include "xla/pjrt/pjrt_common.h"
51#include "xla/pjrt/pjrt_device_description.h"
52#include "xla/pjrt/pjrt_executable.h"
53#include "xla/primitive_util.h"
54#include "xla/shape_util.h"
55#include "xla/tsl/platform/errors.h"
56#include "xla/tsl/platform/logging.h"
57#include "xla/tsl/platform/status.h"
58#include "xla/tsl/platform/statusor.h"
180:26… tr) << "MakeMemoryLayoutDeleter passed PJRT_Api that "
181 "doesn't support layouts extension";
182 return [api, ext_api]( …
577:22… name = absl::string_view(c_value.name, c_value.name_size);
578 switch (c_value.type) {
579 case PJRT_NamedValue_Type::PJRT_NamedValue_kString: {
580 std::string string_value(c_value.string_value, c_value.va …