more lint fix

masahi · Apr 7, 2022 · 07bbb38 · 07bbb38
1 parent 15e60b4
commit 07bbb38
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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