Skip to content

Commit 2def489

Browse files
committed
minor
Signed-off-by: realAsma <[email protected]>
1 parent 56f5a08 commit 2def489

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modelopt/torch/quantization/tensor_quant.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def _fp8_eager(x, amax=None):
5353
x = x.to(torch.float32) * scale_inv
5454
return x.to(dtype)
5555

56+
5657
def fp8_eager(x, amax):
5758
"""Eager mode implementation of FP8 quantization."""
5859
return _fp8_eager(x, amax)

0 commit comments

Comments
 (0)