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 2f13f2d3bf [Unity] Fix importing tvm.contrib.cutlass (#15010)
2f13f2d3bf is described below
commit 2f13f2d3bf99cc369e3e83d0d3e144c949598043
Author: Wuwei Lin <[email protected]>
AuthorDate: Fri Jun 2 16:37:15 2023 +0800
[Unity] Fix importing tvm.contrib.cutlass (#15010)
* [Unity] Fix importing tvm.contrib.cutlass
* Update cutlass.py
---
python/tvm/relax/backend/contrib/cutlass.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/python/tvm/relax/backend/contrib/cutlass.py
b/python/tvm/relax/backend/contrib/cutlass.py
index 323398faef..dffd7c401c 100644
--- a/python/tvm/relax/backend/contrib/cutlass.py
+++ b/python/tvm/relax/backend/contrib/cutlass.py
@@ -21,6 +21,7 @@ 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
@@ -160,7 +161,7 @@ def _check_matmul(context: PatternCheckContext) -> bool:
lhs_shape = lhs.struct_info.shape.values
rhs_shape = rhs.struct_info.shape.values
- return
tvm.contrib.cutlass.build.is_shape_valid_for_cutlass_matmul(lhs_shape,
rhs_shape)
+ return is_shape_valid_for_cutlass_matmul(lhs_shape, rhs_shape)
def _get_activation_from_name(pattern_name):
