Skip to content

Commit ff335a5

Browse files
authored
Update triton import error message (#1842)
1 parent ffb4350 commit ff335a5

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torchao/kernel/intmm.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,10 @@
1414
from torchao.kernel import intmm_triton
1515
else:
1616
intmm_triton = None
17-
except ImportError as e:
18-
logger.debug("import error:", e)
17+
except ImportError:
18+
logger.warning(
19+
"Warning: Detected no triton, on systems without Triton certain kernels will not work"
20+
)
1921
# On cpu-only builds might not be available.
2022
intmm_triton = None
2123

0 commit comments

Comments
 (0)