We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ce20be8 commit 01d0cfeCopy full SHA for 01d0cfe
1 file changed
tests/aiu_addons/test_fp8_addon.py
@@ -13,13 +13,21 @@
13
# limitations under the License.
14
"""Test suite for FMS addon introducing FP8 functionalities"""
15
16
+# Standard
17
+import warnings
18
+
19
# Third Party
20
import pytest
21
import torch
22
23
# Local
24
from fms_mo.prep import available_packages
25
26
+# Suppress the UserWarning about overriding kernel registration in PyTorch 2.8+
27
+# This warning is expected when we override the native CPU kernel for _scaled_mm
28
+warnings.simplefilter("ignore", UserWarning)
29
import fms_mo.aiu_addons.fp8.fp8_spyre_op # pylint: disable=unused-import
30
+warnings.simplefilter("default", UserWarning) # Reset to default after import
31
32
# ============================================================================
33
# Constants
0 commit comments