mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.0 KiB
52 lines
1.0 KiB
{%- macro automodule(modname, options) -%} |
|
.. automodule:: {{ modname }} |
|
{%- for option in options %} |
|
:{{ option }}: |
|
{%- endfor %} |
|
{%- endmacro %} |
|
|
|
{%- macro toctree(docnames) -%} |
|
.. toctree:: |
|
:maxdepth: {{ maxdepth }} |
|
{% for docname in docnames %} |
|
{{ docname }} |
|
{%- endfor %} |
|
{%- endmacro %} |
|
|
|
{%- if is_namespace %} |
|
{{- pkgname | e | heading }} |
|
{% else %} |
|
{{- pkgname | e | heading }} |
|
{% endif %} |
|
|
|
{%- if is_namespace %} |
|
.. py:module:: {{ pkgname }} |
|
{% endif %} |
|
|
|
{%- if modulefirst and not is_namespace %} |
|
{{ automodule(pkgname, automodule_options) }} |
|
{% endif %} |
|
|
|
{%- if subpackages %} |
|
{{ toctree(subpackages) }} |
|
{% endif %} |
|
|
|
{%- if submodules %} |
|
{% if separatemodules %} |
|
{{ toctree(submodules) }} |
|
{% else %} |
|
{%- for submodule in submodules %} |
|
{% if show_headings %} |
|
{{- submodule | e | heading(2) }} |
|
{% endif %} |
|
{{ automodule(submodule, automodule_options) }} |
|
{% endfor %} |
|
{%- endif %} |
|
{%- endif %} |
|
|
|
{%- if not modulefirst and not is_namespace %} |
|
Module contents |
|
--------------- |
|
|
|
{{ automodule(pkgname, automodule_options) }} |
|
{% endif %}
|
|
|