`core.tuning.main`#

Command line for building and inspecting tuned tables.

Usage::

python -m megatron.core.tuning merge  rec.rank*.json -o tables/sm103.json
python -m megatron.core.tuning report rec.rank*.json

Module Contents#

`_out`	Write one line to stdout (`print` is a banned builtin in this repo).
`_err`	Write one line to stderr.
`_load_votes`	Count votes across recordings, or report which file could not be used.
`_merge`	Merge per-rank recordings into a single table file.
`_report`	Summarize recordings without writing a table.
`main`	Entry point for `python -m megatron.core.tuning`.

core.tuning.__main__._out(message: str) → None#: Write one line to stdout (print is a banned builtin in this repo).

core.tuning.__main__._err(message: str) → None#: Write one line to stderr.

core.tuning.__main__._load_votes(records)#: Count votes across recordings, or report which file could not be used.

core.tuning.__main__._merge(args) → int#: Merge per-rank recordings into a single table file.

core.tuning.__main__._report(args) → int#: Summarize recordings without writing a table.

core.tuning.__main__.main(argv=None) → int#: Entry point for python -m megatron.core.tuning.