float8 dynamic autoquant

pytorch · Oct 2, 2024 · ae18023 · ae18023
1 parent 17a1c56
commit ae18023
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
@@ -73,7 +73,7 @@
     AQInt8WeightOnlyQuantizedLinearWeight3,
     AutoQuantizableLinearWeight,
     AQFloat8WeightOnlyQuantizedLinearWeight,
-    AQFloat8DynamicallyQuantizedLinearWeight,
+    AQFloat8PerRowScalingDynamicallyQuantizedLinearWeight,
 )
 from torch.ao.quantization.quantize_fx import convert_to_reference_fx, prepare_fx
 import os