├── .cruft.json ├── .github ├── CONTRIBUTING.md ├── codecov.yml └── workflows │ ├── cruft.yml │ └── tests.yml ├── .gitignore ├── .readthedocs.yml ├── LICENSE ├── README.md ├── docs └── source │ ├── conf.py │ ├── index.rst │ ├── installation.rst │ └── usage.rst ├── pyproject.toml ├── src └── torch_max_mem │ ├── __init__.py │ ├── api.py │ ├── py.typed │ └── version.py ├── tests ├── __init__.py └── test_decorator.py └── tox.ini /.cruft.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.cruft.json -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.github/codecov.yml -------------------------------------------------------------------------------- /.github/workflows/cruft.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.github/workflows/cruft.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/README.md -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/docs/source/installation.rst -------------------------------------------------------------------------------- /docs/source/usage.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/docs/source/usage.rst -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/torch_max_mem/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/src/torch_max_mem/__init__.py -------------------------------------------------------------------------------- /src/torch_max_mem/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/src/torch_max_mem/api.py -------------------------------------------------------------------------------- /src/torch_max_mem/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/torch_max_mem/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/src/torch_max_mem/version.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | """Tests for :mod:`torch_max_mem`.""" 2 | -------------------------------------------------------------------------------- /tests/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/tests/test_decorator.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mberr/torch-max-mem/HEAD/tox.ini --------------------------------------------------------------------------------