pytc.solver.isdf_xtc_ccsd._profile_call

pytc.solver.isdf_xtc_ccsd._profile_call(call, *, schedule_intermediate_estimate_bytes, compiled_memory, rank_panel_size, occupied_pair_batch_size)[source]

Compile once, then record one synchronized steady-state wall time.

Parameters:

  • call (Callable[[], jax.Array])

  • schedule_intermediate_estimate_bytes (int)

  • compiled_memory (CompiledXLAMemory)

  • rank_panel_size (int)

  • occupied_pair_batch_size (int)

Return type:

tuple[jax.Array, FactorDirectProfile]