mirror of https://github.com/hpcaitech/ColossalAI
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.
53 lines
1.0 KiB
53 lines
1.0 KiB
3 years ago
|
{%- 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 %}
|