From cc01d4b8836241e6637ebafedbbf46b0672ef565 Mon Sep 17 00:00:00 2001 From: Jingyan Wang Date: Thu, 25 Jan 2024 22:34:47 +0000 Subject: [PATCH] Give a triton library missing warning instead of silent fail --- orttraining/orttraining/python/training/ortmodule/options.py | 1 + 1 file changed, 1 insertion(+) diff --git a/orttraining/orttraining/python/training/ortmodule/options.py b/orttraining/orttraining/python/training/ortmodule/options.py index df3b078788d16..b9b962acd189b 100644 --- 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