[CUDA] Handle residue k in qmm_naive#3379

zcbenz merged 1 commit intoml-explore:mainml-explore/mlx:mainfrom

zcbenz:qmm-residue-kzcbenz/mlx:qmm-residue-kCopy head branch name to clipboard