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.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