perf: add @torch.compile(dynamic=True) decorator to _grid_search_mse_compiled

Bias92 · Bias92 · commit 243c720031c1 · 2026-03-06T06:31:56.000+09:00
diff --git a/src/llmcompressor/observers/mse.py b/src/llmcompressor/observers/mse.py
@@ -318,6 +318,7 @@ def _grid_search_mse(
     return best_min_val, best_max_val
 
 
+@torch.compile(dynamic=True)
 def _grid_search_mse_compiled(
     observed: torch.Tensor,
     args: QuantizationArgs,
