Merge branch 'main' into nvfuser_slice

Lightning-AI · Feb 3, 2025 · af854f5 · af854f5
2 parents 249b60f + 7f7b8bd
commit af854f5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/thunder/executors/triton_crossentropy.py b/thunder/executors/triton_crossentropy.py
@@ -5,6 +5,12 @@
 
 triton_ex: None | OperatorExecutor = None
 if triton_version is not None:
-    from thunder.executors.triton_crossentropy_impl import triton_ex as impl_ex
+    try:
+        from thunder.executors.triton_crossentropy_impl import triton_ex as impl_ex
 
-    triton_ex = impl_ex
+        triton_ex = impl_ex
+    except Exception:
+        import warnings
+
+        warnings.warn("triton is present but cannot be initialized")
+        triton_version = None