Modules§
Structs§
- Api
- Bool
- Buffer
- Chunk
- Client
- Compile
Options - Compiled
Memory Stats - Copy
ToDevice Stream - Device
- Device
Assignment - Device
Description - Event
- Executable
- Execute
Context - Execute
Options - Execution
- F32
- F64
- I8
- I16
- I32
- I64
- Loaded
Executable - Logical
Id - Memory
- Memory
Stats - Named
Value - Named
Value Map - Program
- Topology
Description - Typed
Host Buffer - U8
- U16
- U32
- U64
Enums§
Traits§
Functions§
Type Aliases§
- Global
Device Id - 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.
- Local
Device Id - 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.
- Local
Hardware Id - 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