pjrt

Classes

Name	Description
`AbstractCompiler`
`ApiMemoryLayout`
`ArtifactDumper`
`BufferInstance`
`ClientInstance`
`CompilerJob`
`CompilerOutput`
`DeviceDescription`
`DeviceInstance`
`DylibPlatform`
`ErrorInstance`
`EventInstance`
`FilesArtifactDumper`
`IREECompiler`
`LoadedExecutableInstance`
`Logger`
`MemoryInstance`
`OpenXLAPartitioner`
`Platform`
`ConfigVars`
`ExecutableImage`
`ResidentExecutable`

Functions

Return	Name	Description
`void`	`BindUndefineds` `static`
`void`	`BindMonomorphicApi`
`PJRT_Error *`	`MakeError` `inline`
`void`	`BindApi` `static`
`std::optional< std::vector< std::string > >`	`ParseOptionsFromCommandLine`
`iree_status_t`	`iree_file_exists` `static`
`void`	`computeBroadcastArgs`

BindUndefineds

static

static void BindUndefineds(PJRT_Api * api)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.cc:2227

BindMonomorphicApi

void BindMonomorphicApi(PJRT_Api * api)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.cc:2241

MakeError

inline

inline PJRT_Error * MakeError(iree_status_t status)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:62

BindApi

static

template<typename PlatformTy, typename ClientInstanceTy> static void BindApi(PJRT_Api * api)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:560

ParseOptionsFromCommandLine

std::optional< std::vector< std::string > > ParseOptionsFromCommandLine(std::string_view options_str)

Defined in integrations/pjrt/src/iree_pjrt/common/command_line_utils.cc:14

iree_file_exists

static

static iree_status_t iree_file_exists(const char * path)

Defined in integrations/pjrt/src/iree_pjrt/common/dylib_platform.cc:177

computeBroadcastArgs

void computeBroadcastArgs(int64_t ndims, int64_t element_size, const int64_t * output_strides, const int64_t * output_shape, int64_t * input_shape, int64_t * perms)

Defined in integrations/pjrt/src/iree_pjrt/common/tensor_utils.cc:15

Variables

Return	Name	Description
`const std::string_view`	`kMlirFormat`
`int64_t`	`kMaxDims` `constexpr`

kMlirFormat

const std::string_view kMlirFormat = "mlir"

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.cc:24

kMaxDims

constexpr

int64_t kMaxDims = 9

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.cc:27