From 07bbb38f7fb52db4a2ecde3d5c87cf4d5cd000a1 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 7 Apr 2022 11:24:56 +0900 Subject: [PATCH] more lint fix --- python/tvm/script/tir/special_stmt.py | 2 +- python/tvm/tir/tensor_intrin/arm_cpu.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tvm/script/tir/special_stmt.py b/python/tvm/script/tir/special_stmt.py index 708acf6aa9a2..3d0fb407ef3f 100644 --- a/python/tvm/script/tir/special_stmt.py +++ b/python/tvm/script/tir/special_stmt.py @@ -840,7 +840,7 @@ def env_thread(env_name, span): self.context.report_error( f"VarDef expected assign to only one var, but got {names}", span ) - v = Var(names[0], span=span) + v = Var(names[0], dtype="int32", span=span) self.context.func_var_env_dict[v] = env_name self.context.update_symbol(v.name, v, self.node) diff --git a/python/tvm/tir/tensor_intrin/arm_cpu.py b/python/tvm/tir/tensor_intrin/arm_cpu.py index a360bf609b18..df658edf323c 100644 --- a/python/tvm/tir/tensor_intrin/arm_cpu.py +++ b/python/tvm/tir/tensor_intrin/arm_cpu.py @@ -15,9 +15,9 @@ # specific language governing permissions and limitations # under the License. # pylint: disable=invalid-name -"""Intrinsics for x86 tensorization.""" -from .. import TensorIntrin +"""Intrinsics for ARM tensorization.""" from tvm.script import tir as T +from .. import TensorIntrin # TODO(masahi): Parametrize the TVMScript description of dot product by