pytc.legacy.kmat._get_safe_batch_size¶

pytc.legacy.kmat._get_safe_batch_size(N_grid, Nb2, dtype=numpy.float64)[source]¶

Calculate safe batch size based on available memory.

Memory requirements for main arrays in calc_K1: - u_grad_batch: (N_grid, batch, 3) - tmp: (Nb^2, batch) Plus some overhead for intermediate calculations