Add float8_static_activation_float8_weight to user-facing API (#994)

pytorch · Oct 2, 2024 · 09b8b3c · 09b8b3c
1 parent e919558
commit 09b8b3c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/torchao/quantization/__init__.py b/torchao/quantization/__init__.py
@@ -46,11 +46,11 @@
     "int4_weight_only",
     "int8_weight_only",
     "uintx_weight_only",
-    "float8_weight_only",
     "fpx_weight_only",
     "LinearActivationQuantizedTensor",
     "to_linear_activation_quantized",
     "to_weight_tensor_with_linear_activation_scale_metadata",
     "float8_weight_only",
-    "float8_dynamic_activation_float8_weight"
+    "float8_dynamic_activation_float8_weight",
+    "float8_static_activation_float8_weight"
 ]
diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py
@@ -93,6 +93,7 @@
     "uintx_weight_only",
     "fpx_weight_only",
     "float8_dynamic_activation_float8_weight",
+    "float8_static_activation_float8_weight",
 ]
 
 from .GPTQ import (