MAINT: linalg: fix up remaining calc_lwork usage