XLA

Reactant.XLA.AllocatorStats Type

AllocatorStats()

Contains the following fields:

• num_allocs

• bytes_in_use

• peak_bytes_in_use

• largest_alloc_size

• bytes_limit

• bytes_reserved

• peak_bytes_reserved

• bytes_reservable_limit

• largest_free_block_bytes

• pool_bytes

• peak_pool_bytes

It should be constructed using the allocatorstats function.

source

Reactant.XLA.allocatorstats Function

allocatorstats([device])

Return an AllocatorStats instance with information about the device specific allocator.

Warning

This method is currently not implemented for the CPU device.

source

Reactant.XLA.claim_gpu_device_mapping! Method

claim_gpu_device_mapping!(owner)

Reserve GPU device mapping configuration for owner. Distributed XLA and MPI/NCCL use incompatible process-to-device setup, so only one of them may configure it.

source

Reactant.XLA.cost_analysis Function

cost_analysis(::AbstractLoadedExecutable)
cost_analysis(::Reactant.Thunk)

Returns a HloCostAnalysisProperties object with the cost analysis of the loaded executable.

source

Reactant.XLA.device_ordinal Function

device_ordinal(device::Device)

Given the device, return the corresponding global device ordinal in the client.

source

Reactant.XLA.device_properties Method

device_properties(device::AbstractDevice)

Get a struct containing device properties. Which exact fields are populated relies on the underlying device implementation.

source

Reactant.XLA.has_initialized_client Method

has_initialized_client(backend)

Return whether a client for backend already exists without triggering lazy backend initialization.

source

Reactant.XLA.isequal_ignoring_metadata Method

isequal_ignoring_metadata(hsharding1::HloSharding, hsharding2::HloSharding)

Same as hsharding1 == hsharding2, but only compares how the data is laid out across the devices. In particular the metadata is ignored, which XLA takes into account through the reduction op (sdy::reduction_op).

source