We ask whether AI is capable of handling the complexity of sparse computation on their own. Unfortunately (or fortnuately?) As of today, the answer is an astounding no. However, we can still leverage AI to augment human expertise.
We present a custom geometric multigrid preconditioned conjugate gradient method that applies summation-by-parts(SBP)-preserving interpolations and a custom matrix-free GPU kernel that achieves up to 5x speedup compared to solvers from PETSc and AmgX