minor

Author: realAsma

modelopt/torch/quantization/tensor_quant.py

@@ -54,9 +54,8 @@ def _fp8_eager(x, amax=None):
     return x.to(dtype)
 
 
-@torch.compile(dynamic=True)
 def _fp8_triton(x, amax):
-    return _fp8_eager(x, amax)
+    return torch.compile(dynamic=True)(_fp8_eager)(x, amax)
 
 
 def fp8_eager(x, amax):