Skip to content

Commit ce20be8

Browse files
committed
lint
Signed-off-by: Andrea Fasoli <andrea.fasoli@ibm.com>
1 parent ca63c7e commit ce20be8

2 files changed

Lines changed: 1 addition & 1 deletion

File tree

fms_mo/aiu_addons/fp8/fp8_linear.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
"""Implement FP8 linear module to be loaded via FMS."""
1515

1616
# Standard
17-
from importlib.metadata import version
1817
from typing import Any, Mapping
1918

2019
# Third Party

fms_mo/aiu_addons/fp8/fp8_spyre_op.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ def _scaled_mm_cpu(
8282
out=None,
8383
)
8484

85+
8586
if torch.__version__ >= "2.8":
8687
# In PyTorch 2.8+, use torch.library.impl to override the native CPU kernel
8788
# The py_kernels dictionary assignment no longer works to override native kernels

0 commit comments

Comments
 (0)