File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 18
18
19
19
from torchao import quantize_
20
20
from torchao .float8 .config import ScalingGranularity
21
- from torchao .float8 .float8_scaling_utils import hp_tensor_to_float8_dynamic
21
+ from torchao .float8 .float8_scaling_utils import _hp_tensor_to_float8_dynamic
22
22
from torchao .float8 .float8_tensor import LinearMMConfig
23
23
from torchao .quantization .granularity import (
24
24
PerAxis ,
@@ -1703,7 +1703,7 @@ def test_float8_rowwise_fake_quantize(self):
1703
1703
x = torch .randn (32 , 64 )
1704
1704
axiswise_dim = 0
1705
1705
out = _Float8RowwiseFakeQuantize .apply (x , dtype , axiswise_dim )
1706
- out_expected = hp_tensor_to_float8_dynamic (
1706
+ out_expected = _hp_tensor_to_float8_dynamic (
1707
1707
x ,
1708
1708
dtype ,
1709
1709
LinearMMConfig (),
You can’t perform that action at this time.
0 commit comments