tilelang.tools.cuda.iket.cli

Host-side output and IKET CLI helpers.

Functions

set_output_dir(path)	Set the process-local IKET export directory used by helper APIs.
output_dir([default])	Return the configured IKET export directory, if any.
output_path(name, *[, directory])	Return a path under the configured IKET export directory.
trace_files(*[, directory])	Return generated IKET JSON traces in size-descending order.
profile_command(command, *, directory[, postprocess, ...])	Build an IKET CLI command that exports traces to directory.

Module Contents

tilelang.tools.cuda.iket.cli.set_output_dir(path)

Set the process-local IKET export directory used by helper APIs.

Parameters:

path (str | os.PathLike[str] | None)

Return type:

pathlib.Path | None

tilelang.tools.cuda.iket.cli.output_dir(default=None)

Return the configured IKET export directory, if any.

Parameters:

default (str | os.PathLike[str] | None)

Return type:

pathlib.Path | None

tilelang.tools.cuda.iket.cli.output_path(name, *, directory=None)

Return a path under the configured IKET export directory.

Parameters:

• name (str)

• directory (str | os.PathLike[str] | None)

Return type:

pathlib.Path

tilelang.tools.cuda.iket.cli.trace_files(*, directory=None)

Return generated IKET JSON traces in size-descending order.

Parameters:

directory (str | os.PathLike[str] | None)

Return type:

list[pathlib.Path]

tilelang.tools.cuda.iket.cli.profile_command(command, *, directory, postprocess='all', clobber=True)

Build an IKET CLI command that exports traces to directory.

Parameters:

• command (str | collections.abc.Sequence[str])

• directory (str | os.PathLike[str])

• postprocess (str)

• clobber (bool)

Return type:

str