tilelang.ascend.host_profile¶

Optional Torch host ranges for Ascend kernel calls.

Functions¶

maybe_wrap_host_profile(func, prim_func)

Optionally wrap an Ascend callable in a Torch host profiling range.

Module Contents¶

tilelang.ascend.host_profile.maybe_wrap_host_profile(func, prim_func)¶

Optionally wrap an Ascend callable in a Torch host profiling range.

Resolve the switch and range name once when the callable is created or restored from disk. Disabled profiling and missing global_symbol attributes leave the original callable unchanged.

Parameters:

  • func (collections.abc.Callable[Ellipsis, Any])

  • prim_func (tvm.tirx.PrimFunc)

Return type:

collections.abc.Callable[Ellipsis, Any]