├── .gitignore ├── CHANGELOG.md ├── CITATION.cff ├── CONTRIBUTING.md ├── LICENSE_code.md ├── LICENSE_doc.md ├── MANIFEST.in ├── README.md ├── doc └── img │ ├── algorithm.png │ ├── logo.png │ └── logo_small.png ├── package.sh ├── pyproject.toml ├── requirements.txt ├── setup.py ├── tests ├── requirements_test.txt └── test.py └── torch_discounted_cumsum ├── __init__.py ├── discounted_cumsum.py ├── discounted_cumsum_cpu.cpp ├── discounted_cumsum_cuda.cpp ├── discounted_cumsum_cuda_kernel.cu └── version.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea 3 | *.egg-info 4 | venv* 5 | deploy* 6 | -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE_code.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/LICENSE_code.md -------------------------------------------------------------------------------- /LICENSE_doc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/LICENSE_doc.md -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/README.md -------------------------------------------------------------------------------- /doc/img/algorithm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/doc/img/algorithm.png -------------------------------------------------------------------------------- /doc/img/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/doc/img/logo.png -------------------------------------------------------------------------------- /doc/img/logo_small.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/doc/img/logo_small.png -------------------------------------------------------------------------------- /package.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/package.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.5 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/setup.py -------------------------------------------------------------------------------- /tests/requirements_test.txt: -------------------------------------------------------------------------------- 1 | torch>=1.5 2 | tqdm 3 | -------------------------------------------------------------------------------- /tests/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/tests/test.py -------------------------------------------------------------------------------- /torch_discounted_cumsum/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/torch_discounted_cumsum/__init__.py -------------------------------------------------------------------------------- /torch_discounted_cumsum/discounted_cumsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/torch_discounted_cumsum/discounted_cumsum.py -------------------------------------------------------------------------------- /torch_discounted_cumsum/discounted_cumsum_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/torch_discounted_cumsum/discounted_cumsum_cpu.cpp -------------------------------------------------------------------------------- /torch_discounted_cumsum/discounted_cumsum_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/torch_discounted_cumsum/discounted_cumsum_cuda.cpp -------------------------------------------------------------------------------- /torch_discounted_cumsum/discounted_cumsum_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/torch-discounted-cumsum/HEAD/torch_discounted_cumsum/discounted_cumsum_cuda_kernel.cu -------------------------------------------------------------------------------- /torch_discounted_cumsum/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '1.1.0' --------------------------------------------------------------------------------