Skip to content

Commit

Permalink
[TIR][USMP] adding the pass to convert to pool offsets
Browse files Browse the repository at this point in the history
* fixing typos

Change-Id: I42c557fd394aefdf8c2e825c4e88770eb0732f9b
  • Loading branch information
manupak committed Dec 8, 2021
1 parent 52a4f12 commit 9ce59ee
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def _get_primfuncs_from_module(module):


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

def set_poolinfos(stmt):
if isinstance(stmt, tvm.tir.Allocate):
Expand All @@ -49,7 +49,7 @@ def set_poolinfos(stmt):


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


def _assign_targets_to_primfuncs_irmodule(mod, target):
"""helper to assign target for PrimFunc in a IRModule"""
"""Helper to assign target for PrimFunc in a IRModule"""
ret = tvm.IRModule()
for global_var, basefunc in mod.functions.items():
if isinstance(basefunc, tvm.tir.PrimFunc):
Expand Down

0 comments on commit 9ce59ee

Please sign in to comment.