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