removed torch.compile to fix onnx unittests

Signed-off-by: realAsma <[email protected]>

Author: realAsma

modelopt/torch/quantization/algorithms.py

@@ -668,12 +668,10 @@ def run_search(self):
         QuantRecipe.fold_pqs_to_weights(self.model)
 
 
-@torch.compile
 def _get_auto_quantize_score(grad_output, output_diff):
     return ((grad_output.float() ** 2) * (output_diff.float() ** 2)).sum()
 
 
-@torch.compile
 def _add_auto_quantize_score(grad_output, output_diff, score_tensor):
     score_tensor += _get_auto_quantize_score(grad_output, output_diff)