fixed fp8 import so as to not require user to import

JRosenkranz · JRosenkranz · commit 57cd6f63bfcc · 2025-08-06T19:44:30.000Z
Signed-off-by: Joshua Rosenkranz &lt;jmrosenk@us.ibm.com&gt;
diff --git a/fms_mo/aiu_addons/__init__.py b/fms_mo/aiu_addons/__init__.py
@@ -30,6 +30,7 @@ def _infer_quantization_config(quant_config: dict) -> dict | None:
                 )
             # First, import required FP8 linear classes from fms-mo
             # Local
+            import fms_mo.aiu_addons.fp8.fp8_attn  # pylint: disable=unused-import
             import fms_mo.aiu_addons.fp8.fp8_adapter  # pylint: disable=unused-import
             import fms_mo.aiu_addons.fp8.fp8_linear  # pylint: disable=unused-import
 

Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ def _infer_quantization_config(quant_config: dict) -> dict \| None:`
`30`	`30`	`)`
`31`	`31`	`# First, import required FP8 linear classes from fms-mo`
`32`	`32`	`# Local`
	`33`	`+ import fms_mo.aiu_addons.fp8.fp8_attn # pylint: disable=unused-import`
`33`	`34`	`import fms_mo.aiu_addons.fp8.fp8_adapter # pylint: disable=unused-import`
`34`	`35`	`import fms_mo.aiu_addons.fp8.fp8_linear # pylint: disable=unused-import`
`35`	`36`