torch_cg icon indicating copy to clipboard operation
torch_cg copied to clipboard

Only call A_bmm once per iter, rather than three times

Open PMKielstra opened this issue 5 months ago • 0 comments

In general, the most expensive part of any CG solve is computing A_bmm. This version will do it once per iteration rather than three times.

PMKielstra avatar Aug 26 '24 23:08 PMKielstra