diff --git a/3rdparty/tvm b/3rdparty/tvm index 8dff258d2..bcdcec2c7 160000 --- a/3rdparty/tvm +++ b/3rdparty/tvm @@ -1 +1 @@ -Subproject commit 8dff258d2837b2c0d24619ebf26dd596b2291912 +Subproject commit bcdcec2c7502a565ea724fc145eebc993eebc484 diff --git a/bitblas/gpu/matmul_mma.py b/bitblas/gpu/matmul_mma.py index 4bf8be4e6..629d93768 100644 --- a/bitblas/gpu/matmul_mma.py +++ b/bitblas/gpu/matmul_mma.py @@ -8,13 +8,13 @@ from tvm import tir, DataType from tvm.target import Target -from ..base.roller import Hint -from ..base.roller.rasterization import NoRasterization -from ..base import analysis -from .base import GPUScheduleRule -from .matmul_mma_dequantize import MatmulTensorizationMMAWithDequantizeInfo -from ..base.analysis import get_coalesced_veclen -from .matmul_analysis import ( +from bitblas.base.roller import Hint +from bitblas.base.roller.rasterization import NoRasterization +from bitblas.base import analysis +from bitblas.gpu.base import GPUScheduleRule +from bitblas.gpu.matmul_mma_dequantize import MatmulTensorizationMMAWithDequantizeInfo +from bitblas.base.analysis import get_coalesced_veclen +from bitblas.gpu.matmul_analysis import ( auto_inline_consumer_chain, is_transpose_block, is_identity_block, diff --git a/bitblas/gpu/matmul_mma_dequantize.py b/bitblas/gpu/matmul_mma_dequantize.py index de1b5b896..3a8cf048a 100644 --- a/bitblas/gpu/matmul_mma_dequantize.py +++ b/bitblas/gpu/matmul_mma_dequantize.py @@ -8,13 +8,13 @@ from tvm import tir, DataType -from ..base.roller.hint import Hint, IntrinInfo +from bitblas.base.roller.hint import Hint, IntrinInfo from tvm.target import Target -from ..base.roller.rasterization import NoRasterization -from ..base import analysis -from .base import GPUScheduleRule -from ..base.analysis import get_coalesced_veclen -from .matmul_analysis import ( +from bitblas.base.roller.rasterization import NoRasterization +from bitblas.base import analysis +from bitblas.gpu.base import GPUScheduleRule +from bitblas.base.analysis import get_coalesced_veclen +from bitblas.gpu.matmul_analysis import ( auto_inline_consumer_chain, auto_inline_producers, get_reduction_blocks,