We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 320024e commit 88bb2c3Copy full SHA for 88bb2c3
torchao/kernel/intmm.py
@@ -1,17 +1,21 @@
1
+import logging
2
import os
3
4
import torch
5
6
from torchao.utils import TORCH_VERSION_AT_LEAST_2_2, TORCH_VERSION_AT_LEAST_2_6
7
8
+logger = logging.getLogger(__name__)
9
+logger.addHandler(logging.NullHandler())
10
+
11
try:
12
# Only works for torch2.2 or newer.
13
if TORCH_VERSION_AT_LEAST_2_2:
14
from torchao.kernel import intmm_triton
15
else:
16
intmm_triton = None
17
except ImportError as e:
- print("import error:", e)
18
+ logger.debug("import error:", e)
19
# On cpu-only builds might not be available.
20
21
0 commit comments