Crate pjrt

Source

Modules§

protos

Structs§

Api
Bool
Buffer
Chunk
Client
CompileOptions
CompiledMemoryStats
CopyToDeviceStream
Device
DeviceAssignment
DeviceDescription
Event
Executable
ExecuteContext
ExecuteOptions
Execution
F32
F64
I8
I16
I32
I64
LoadedExecutable
LogicalId
Memory
MemoryStats
NamedValue
NamedValueMap
Program
TopologyDescription
TypedHostBuffer
U8
U16
U32
U64

Enums§

Error
HostBuffer
MemoryLayout
PrimitiveType
ProgramFormat

Traits§

AsDType
CompileToExecutable
CompileToLoadedExecutable
DType
ElemType
ExecutionInputs
KeyValueStore
Type

Functions§

plugin

Type Aliases§

GlobalDeviceId
The logical global device ID. This is unique among devices of this type (e.g. CPUs, GPUs). On multi-host platforms, this will be unique across all hosts’ devices.
LocalDeviceId
The logical local device ID. This will be used to look up an addressable device local to a given client. It is -1 if undefined.
LocalHardwareId
The physical local device ID, e.g., the CUDA device number. Multiple PJRT devices can have the same LocalHardwareId if these PJRT devices share the same physical device.In general, not guaranteed to be dense, and -1 if undefined.
Result