Skip to content

Commit 07bbb38

Browse files
committed
more lint fix
1 parent 15e60b4 commit 07bbb38

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/tvm/script/tir/special_stmt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -840,7 +840,7 @@ def env_thread(env_name, span):
840840
self.context.report_error(
841841
f"VarDef expected assign to only one var, but got {names}", span
842842
)
843-
v = Var(names[0], span=span)
843+
v = Var(names[0], dtype="int32", span=span)
844844
self.context.func_var_env_dict[v] = env_name
845845
self.context.update_symbol(v.name, v, self.node)
846846

python/tvm/tir/tensor_intrin/arm_cpu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
# specific language governing permissions and limitations
1616
# under the License.
1717
# pylint: disable=invalid-name
18-
"""Intrinsics for x86 tensorization."""
19-
from .. import TensorIntrin
18+
"""Intrinsics for ARM tensorization."""
2019
from tvm.script import tir as T
20+
from .. import TensorIntrin
2121

2222

2323
# TODO(masahi): Parametrize the TVMScript description of dot product by

0 commit comments

Comments
 (0)