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