Skip to content

Commit 95ab806

Browse files
committed
fix linter
1 parent f28f02f commit 95ab806

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/tir/transforms/memhammer_rewrite_rule.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919
#ifndef TVM_TIR_TRANSFORMS_MEMHAMMER_REWRITE_RULE_H_
2020
#define TVM_TIR_TRANSFORMS_MEMHAMMER_REWRITE_RULE_H_
2121

22-
2322
#include <tvm/arith/iter_affine_map.h>
2423
#include <tvm/runtime/registry.h>
2524
#include <tvm/target/target.h>

tests/python/unittest/test_tir_transform_memhammer_lower_auto_copy.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1031,11 +1031,13 @@ def verify(n):
10311031
assert num_alloc[0] == 1
10321032

10331033
if alloc_size:
1034+
10341035
def prod(arr):
10351036
ret = 1
10361037
for element in arr:
10371038
ret *= element
10381039
return ret
1040+
10391041
assert prod(alloc_extents[0]) == alloc_size
10401042

10411043

0 commit comments

Comments
 (0)