pytc.solver.isdf_xtc_ccsd.compiled_full_thc_memory

pytc.solver.isdf_xtc_ccsd.compiled_full_thc_memory(t2, a, b, z, c, e, *, occupied_pair_batch_size=8, rank_panel_size=128)[source]

Return XLA accounting for the compiled full-THC branch executable.

Parameters:
  • t2 (jax.Array)

  • a (jax.Array)

  • b (jax.Array)

  • z (jax.Array)

  • c (jax.Array)

  • e (jax.Array)

  • occupied_pair_batch_size (int)

  • rank_panel_size (int)

Return type:

CompiledXLAMemory