pytc.solver.xtc_ccsd._TiledBlockSpec

class pytc.solver.xtc_ccsd._TiledBlockSpec(name, ranges_fn, panel_layout, trim_fn, df_fn, write_fn)

Bases: object

Describes one ERI block built tile-by-tile via the multi-GPU pipeline.

All per-block logic (index ranges, padding trim, DF contribution, output accumulation) lives here so _run_tiled_block_pipeline contains only the device-dispatch boilerplate.

Parameters:

• name (str) – Human-readable block name used in log messages.

• ranges_fn (callable (i0, i1) -> tuple[slice, ...]) – Returns the four MO-index slices for tile [i0:i1].

• panel_layout (str) – "pr" or "qr" — which axes are JIT-padded to panel_size.

• trim_fn (callable (tc_raw, i_len) -> ndarray) – Strips XTC padding to the actual tile shape.

• df_fn (callable (i0, i1) -> ndarray or None) – CPU DF contribution for the tile (same shape as trimmed XTC result). Pass None to skip the DF addition.

• write_fn (callable (i0, i1, tile) -> None) – Accumulates / writes the finished tile. Called under acc_lock.

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

name
ranges_fn
panel_layout
trim_fn
df_fn
write_fn
__annotations__
__doc__
__module__
__slots__

name

ranges_fn

panel_layout

trim_fn

df_fn

write_fn