pytc.solver.xtc_ccsd._compute_large_blocks¶
- pytc.solver.xtc_ccsd._compute_large_blocks(eris, xtc_obj, jastrow_params, Lov_reshaped, L_vv_full, nocc, nvir, nmo, panel_blk)[source]¶
Compute and write balanced tiled ovvv / vovv blocks to HDF5.
The r virtual index is sliced in chunks of
panel_blk(GPU-memory-aware, fromresolve_v3o_panel_block_size). Both tensors are tiled in a single interleaved_run_tiled_block_pipelinecall so GPUs stay occupied across block boundaries.Both tensors are written as [:, :, r0:r1, :] slabs on axis 2.
ovvv tile: (occ_all, vir_all, vir_r_blk, vir_all) panel_layout=”pr” vovv tile: (vir_all, occ_all, vir_r_blk, vir_all) panel_layout=”qr”
panel_size=max(nocc, panel_blk) pads the variable-sized dimensions (p=occ for ovvv, q=occ for vovv, and r=vir_r_blk) to a fixed size for JIT shape stability. The large vir_all dimensions remain fixed across all tiles.