Fix import pytorch error

Signed-off-by: wenjh <wenjh@sugon.com>

Fix import pytorch error
Signed-off-by: wenjh <wenjh@sugon.com>
3939e719 · wenjh · 3b1f30a9 · 3939e719
Commit 3939e719 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
@@ -12,7 +12,10 @@ import transformer_engine.common
 try:
    from . import pytorch
 except ImportError as e:
-    pass
+    try:
+        from . import pytorch
+    except ImportError as e:
+        pass

 try:
    from . import jax