From 4289d7bb604f3a73c3a37252fea91ef734414237 Mon Sep 17 00:00:00 2001 From: LeiWang1999 Date: Fri, 9 Aug 2024 16:34:00 +0000 Subject: [PATCH] dsl fix --- docs/ExtendOperatorsWithDSL.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/ExtendOperatorsWithDSL.md b/docs/ExtendOperatorsWithDSL.md index 1c01602b..279cc249 100644 --- a/docs/ExtendOperatorsWithDSL.md +++ b/docs/ExtendOperatorsWithDSL.md @@ -1,7 +1,7 @@ ### Using BitBLAS from DSL ```python from bitblas.base.roller.policy import TensorCorePolicy, DefaultPolicy -from bitblas.base.roller.arch import CUDA +from bitblas.base.arch import CUDA from bitblas.base.utils import apply_and_build @tvm.script.ir_module class MatmulNT: