pytc.solver.isdf_xtc_ccsd.CompiledXLAMemory

class pytc.solver.isdf_xtc_ccsd.CompiledXLAMemory(temporary_bytes, argument_bytes, output_bytes, alias_bytes)[source]

Bases: object

Backend-specific buffer accounting reported by a JAX executable.

Methods

__delattr__

Implement delattr(self, name).

__dir__

Default dir() implementation.

__eq__

Return self==value.

__format__

Default object formatter.

__ge__

Return self>=value.

__getattribute__

Return getattr(self, name).

__getstate__

Helper for pickle.

__gt__

Return self>value.

__hash__

Return hash(self).

__init__

__init_subclass__

This method is called when a class is subclassed.

__le__

Return self<=value.

__lt__

Return self<value.

__ne__

Return self!=value.

__new__

__reduce__

Helper for pickle.

__reduce_ex__

Helper for pickle.

__repr__

Return repr(self).

__setattr__

Implement setattr(self, name, value).

__sizeof__

Size of object in memory, in bytes.

__str__

Return str(self).

__subclasshook__

Abstract classes can override this to customize issubclass().

Attributes

__annotations__

__dataclass_fields__

__dataclass_params__

__dict__

__doc__

__match_args__

__module__

__weakref__

list of weak references to the object

total_bytes

XLA's argument + output + temporary accounting, less aliases.

temporary_bytes

argument_bytes

output_bytes

alias_bytes

Parameters:
  • temporary_bytes (int)

  • argument_bytes (int)

  • output_bytes (int)

  • alias_bytes (int)

temporary_bytes: int
argument_bytes: int
output_bytes: int
alias_bytes: int
property total_bytes: int

XLA’s argument + output + temporary accounting, less aliases.