Skip to content

Commit 9ce59ee

Browse files
committed
[TIR][USMP] adding the pass to convert to pool offsets
* fixing typos Change-Id: I42c557fd394aefdf8c2e825c4e88770eb0732f9b
1 parent 52a4f12 commit 9ce59ee

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/python/unittest/test_tir_usmp_transform_convert_pool_allocations_to_offsets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def _get_primfuncs_from_module(module):
3232

3333

3434
def assign_poolinfos_to_allocates_in_primfunc(primfunc, pool_infos):
35-
"""helper to assing poolinfos to allocate nodes in a tir.PrimFunc"""
35+
"""Helper to assign poolinfos to allocate nodes in a tir.PrimFunc"""
3636

3737
def set_poolinfos(stmt):
3838
if isinstance(stmt, tvm.tir.Allocate):
@@ -49,7 +49,7 @@ def set_poolinfos(stmt):
4949

5050

5151
def assign_poolinfos_to_allocates_in_irmodule(mod, pool_infos):
52-
"""helper to assing poolinfos to allocate nodes in a IRModule"""
52+
"""Helper to assign poolinfos to allocate nodes in a IRModule"""
5353
ret = tvm.IRModule()
5454
for global_var, basefunc in mod.functions.items():
5555
if isinstance(basefunc, tvm.tir.PrimFunc):
@@ -58,7 +58,7 @@ def assign_poolinfos_to_allocates_in_irmodule(mod, pool_infos):
5858

5959

6060
def _assign_targets_to_primfuncs_irmodule(mod, target):
61-
"""helper to assign target for PrimFunc in a IRModule"""
61+
"""Helper to assign target for PrimFunc in a IRModule"""
6262
ret = tvm.IRModule()
6363
for global_var, basefunc in mod.functions.items():
6464
if isinstance(basefunc, tvm.tir.PrimFunc):

0 commit comments

Comments
 (0)