Skip to content

Commit 96740bc

Browse files
author
shingjan
committed
add exp test and mypy ignore
1 parent 5a2d54f commit 96740bc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/python/unittest/test_tvmscript_type.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def lowered_loop_split(a: T.handle, b: T.handle) -> None:
149149

150150

151151
"""
152-
This test case is added to test T.Buffer with slice as argument
152+
This test case is added to test T.Buffer with slice as argument and T.exp
153153
"""
154154

155155

@@ -165,13 +165,13 @@ def different_access_indices(a: T.handle, b: T.handle) -> None:
165165
T.writes(
166166
[
167167
B[
168-
T.min(vj, vi) : T.min(vj, vi) + (T.max(vj, vi) + 1 - T.min(vj, vi)),
169-
T.min(vi, vj) : T.min(vi, vj) + (T.max(vi, vj) + 1 - T.min(vi, vj)),
168+
T.min(vj, vi) : T.min(vj, vi) + (T.max(vj, vi) + 1 - T.min(vj, vi)), # type: ignore[misc]
169+
T.min(vi, vj) : T.min(vi, vj) + (T.max(vi, vj) + 1 - T.min(vi, vj)), # type: ignore[misc]
170170
]
171171
]
172172
)
173173
with T.init():
174-
B[vj, vi] = T.float32(0)
174+
B[vj, vi] = T.exp(B[vi, vj])
175175
B[vi, vj] = B[vi, vj] + A[vi, vj, vk]
176176

177177

0 commit comments

Comments
 (0)