From 9df46e7e7e196536befe1e17b622938a90dde27e Mon Sep 17 00:00:00 2001 From: Gen TANG Date: Sun, 26 Jul 2026 22:25:57 +0800 Subject: [PATCH 1/2] fix tilelang dependency bug --- tile_kernels/quant/common.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tile_kernels/quant/common.py b/tile_kernels/quant/common.py index 33f72b0..f09b8af 100644 --- a/tile_kernels/quant/common.py +++ b/tile_kernels/quant/common.py @@ -4,7 +4,12 @@ import torch from tilelang import language as T from tilelang.contrib import nvcc -from tilelang.utils.target import determine_target +try: + # TileLang >= 0.1.12 + from tilelang.backend.target import determine_target +except ModuleNotFoundError: + # TileLang <= 0.1.11 + from tilelang.utils.target import determine_target from tile_kernels.quant.types import QuantTensor from tile_kernels.utils import align, ceil_div From 251c38fa7273ff25e09e85df3b5263c52e7f19fd Mon Sep 17 00:00:00 2001 From: Gen TANG Date: Mon, 27 Jul 2026 09:14:03 +0800 Subject: [PATCH 2/2] catch ImportError instead of ModuleNotFoundError --- tile_kernels/quant/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tile_kernels/quant/common.py b/tile_kernels/quant/common.py index f09b8af..a057b5c 100644 --- a/tile_kernels/quant/common.py +++ b/tile_kernels/quant/common.py @@ -7,7 +7,7 @@ try: # TileLang >= 0.1.12 from tilelang.backend.target import determine_target -except ModuleNotFoundError: +except ImportError: # TileLang <= 0.1.11 from tilelang.utils.target import determine_target