FlashMatrix is mainly optimized for data matrix (n >> p or n << p). We can fix n * p, and vary n and p to see how well FlashMatrix can perform. We can use ridge regression to evaluate its performance.