Give a triton library missing warning instead of silent fail

microsoft · Jan 25, 2024 · cc01d4b · cc01d4b
1 parent a2867b9
commit cc01d4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/orttraining/orttraining/python/training/ortmodule/options.py b/orttraining/orttraining/python/training/ortmodule/options.py
@@ -379,6 +379,7 @@ def _override_from_env_vars(self):
                 import triton  # noqa: F401
             except ImportError:
                 pass
+                self._logger.warning("triton is not found. Please install triton with `pip install triton`")
             else:
                 self.enable_triton = True