From c6ab0ced289af188ac29b3520f7a3ae11274fba5 Mon Sep 17 00:00:00 2001 From: Wenhua Cheng Date: Wed, 10 Sep 2025 09:28:44 +0800 Subject: [PATCH] remove torch compile --- auto_round/data_type/nvfp.py | 1 - 1 file changed, 1 deletion(-) diff --git a/auto_round/data_type/nvfp.py b/auto_round/data_type/nvfp.py index 12e8e4bfb..5d7e6221f 100644 --- a/auto_round/data_type/nvfp.py +++ b/auto_round/data_type/nvfp.py @@ -67,7 +67,6 @@ def calculate_gparam(tensor, group_size=16, device="cpu"): return global_scale -@torch.compile() def ref_nvfp4_quant(x, global_scale, block_size=16, v=0, scale_coeff=1.0): assert global_scale.dtype == torch.float32 assert x.ndim == 2