From c513c47fc5db13762009efbc96461fb95ddcb3c8 Mon Sep 17 00:00:00 2001 From: scavallari Date: Mon, 8 Dec 2025 02:46:46 -0800 Subject: [PATCH] us gesvd as solver Signed-off-by: scavallari --- modelopt/torch/quantization/model_calib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelopt/torch/quantization/model_calib.py b/modelopt/torch/quantization/model_calib.py index d4cf249fe..1e18dbf07 100644 --- a/modelopt/torch/quantization/model_calib.py +++ b/modelopt/torch/quantization/model_calib.py @@ -1033,7 +1033,7 @@ def postprocess(module, name): original_device = weight.device original_dtype = weight.dtype weight_f64 = weight.to(dtype=torch.float64, device=original_device) - u, s, vt = torch.linalg.svd(weight_f64, full_matrices=False) + u, s, vt = torch.linalg.svd(weight_f64, driver="gesvd", full_matrices=False) if u.shape[1] < lowrank or vt.shape[0] < lowrank: warnings.warn( "The low-rank dimensions do not match the layer dimensions. "