This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/unity by this push:
new f360556a0f [Unity][CUTLASS] Fix circular import bug in relax cutlass
backend (#15001)
f360556a0f is described below
commit f360556a0ff39de254d0d37002bc4685bc3bc5ad
Author: Nomadc <[email protected]>
AuthorDate: Thu Jun 1 21:12:40 2023 +0200
[Unity][CUTLASS] Fix circular import bug in relax cutlass backend (#15001)
* [Unity][CUTLASS] Fix circular import bug
* [Unity][CUTLASS] remove unnecessary cutlass import.
* [Unity][CUTLASS] remove not used __init__ file.
---
python/tvm/relax/backend/contrib/__init__.py | 20 --------------------
python/tvm/relax/backend/contrib/cutlass.py | 3 +--
2 files changed, 1 insertion(+), 22 deletions(-)
diff --git a/python/tvm/relax/backend/contrib/__init__.py
b/python/tvm/relax/backend/contrib/__init__.py
deleted file mode 100644
index a094c97d24..0000000000
--- a/python/tvm/relax/backend/contrib/__init__.py
+++ /dev/null
@@ -1,20 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-"""External backend codegen modules for Relax."""
-
-from .cutlass import partition_for_cutlass
diff --git a/python/tvm/relax/backend/contrib/cutlass.py
b/python/tvm/relax/backend/contrib/cutlass.py
index dffd7c401c..323398faef 100644
--- a/python/tvm/relax/backend/contrib/cutlass.py
+++ b/python/tvm/relax/backend/contrib/cutlass.py
@@ -21,7 +21,6 @@ from functools import reduce
from typing import Mapping, Sequence
import tvm
-from tvm.contrib.cutlass.build import is_shape_valid_for_cutlass_matmul
from tvm.relax import Call, DataflowVar, Function, PyExprMutator, Var,
expr_functor, transform
from tvm.relax.dpl import rewrite_call
from tvm.relax.transform import PatternCheckContext
@@ -161,7 +160,7 @@ def _check_matmul(context: PatternCheckContext) -> bool:
lhs_shape = lhs.struct_info.shape.values
rhs_shape = rhs.struct_info.shape.values
- return is_shape_valid_for_cutlass_matmul(lhs_shape, rhs_shape)
+ return
tvm.contrib.cutlass.build.is_shape_valid_for_cutlass_matmul(lhs_shape,
rhs_shape)
def _get_activation_from_name(pattern_name):
