add warning suppression for op override

Signed-off-by: Andrea Fasoli <andrea.fasoli@ibm.com>

Author: andrea-fasoli

fms_mo/aiu_addons/fp8/fp8_spyre_op.py

@@ -15,6 +15,7 @@
 
 # Standard
 from typing import Optional
+import warnings
 
 # Third Party
 from torch import Tensor
@@ -87,8 +88,15 @@ def _scaled_mm_cpu(
     # In PyTorch 2.8+, use torch.library.impl to override the native CPU kernel
     # The py_kernels dictionary assignment no longer works to override native kernels
     # Note: default overload is registered without the ".default" suffix
-    torch.library.impl("aten::_scaled_mm.out", "CPU")(_scaled_mm_cpu_out)
-    torch.library.impl("aten::_scaled_mm", "CPU")(_scaled_mm_cpu)
+    # Suppress the UserWarning about overriding a previously registered kernel
+    with warnings.catch_warnings():
+        warnings.filterwarnings(
+            "ignore",
+            category=UserWarning,
+            message=".*Overriding a previously registered kernel.*",
+        )
+        torch.library.impl("aten::_scaled_mm.out", "CPU")(_scaled_mm_cpu_out)
+        torch.library.impl("aten::_scaled_mm", "CPU")(_scaled_mm_cpu)
 else:
     torch.library.register_kernel(
         torch.ops.aten._scaled_mm.out, "cpu", _scaled_mm_cpu_out