diff --git a/tile_kernels/quant/common.py b/tile_kernels/quant/common.py index 33f72b0..a057b5c 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 ImportError: + # 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