From 75263abbcdf51fd8e5e698bda7ab465aeccabd07 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Sat, 1 Mar 2025 12:47:56 -0800 Subject: [PATCH 1/2] Silence loud commit --- torchao/kernel/intmm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/torchao/kernel/intmm.py b/torchao/kernel/intmm.py index 9448fcce59..c3b2a3d9be 100644 --- a/torchao/kernel/intmm.py +++ b/torchao/kernel/intmm.py @@ -1,9 +1,13 @@ +import logging import os import torch from torchao.utils import TORCH_VERSION_AT_LEAST_2_2, TORCH_VERSION_AT_LEAST_2_6 +logger = logging.getLogger(__name__) + + try: # Only works for torch2.2 or newer. if TORCH_VERSION_AT_LEAST_2_2: @@ -11,7 +15,7 @@ else: intmm_triton = None except ImportError as e: - print("import error:", e) + logger.debug("import error:", e) # On cpu-only builds might not be available. intmm_triton = None From 77e2b3f8158dc49b60b14c4451a5b62192638ddc Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Mon, 3 Mar 2025 12:28:54 -0800 Subject: [PATCH 2/2] Update intmm.py --- torchao/kernel/intmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchao/kernel/intmm.py b/torchao/kernel/intmm.py index c3b2a3d9be..b460b04dea 100644 --- a/torchao/kernel/intmm.py +++ b/torchao/kernel/intmm.py @@ -6,7 +6,7 @@ from torchao.utils import TORCH_VERSION_AT_LEAST_2_2, TORCH_VERSION_AT_LEAST_2_6 logger = logging.getLogger(__name__) - +logger.addHandler(logging.NullHandler()) try: # Only works for torch2.2 or newer.