Merge branch 'develop_v2.4'

f9faa7ca · wenjh · fdf60506 · 3939e719 · f9faa7ca
Commit f9faa7ca authored Jul 15, 2025 by wenjh
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

transformer_engine/__init__.py transformer_engine/__init__.py +4 -1

No files found.
--- a/transformer_engine/__init__.py
+++ b/transformer_engine/__init__.py
@@ -13,7 +13,10 @@ import transformer_engine.common
 try:
    from . import pytorch
 except ImportError:
-    pass
+    try:
+        from . import pytorch
+    except ImportError:
+        pass
 except FileNotFoundError as e:
    if "Could not find shared object file" not in str(e):
        raise e  # Unexpected error