/td3a_cpp_deep

Example of a python module using cpp and torch

Primary LanguagePythonMIT LicenseMIT

https://circleci.com/gh/sdpython/td3a_cpp_deep/tree/main.svg?style=svg Build status Build Status Windows https://dev.azure.com/xavierdupre3/td3a_cpp_deep/_apis/build/status/sdpython.td3a_cpp_deep GitHub Issues MIT License size

td3a_cpp_deep: template to implement torch extension

https://raw.githubusercontent.com/sdpython/td3a_cpp_deep/main/doc/_static/logo.png

documentation

Simple template to implement an algorithm with cython and openmp. It implements simple examples to demonstrate the speed up obtained by using cython. The module must be compiled to be used inplace:

python setup.py build_ext --inplace

Generate the setup in subfolder dist:

python setup.py sdist

Generate the documentation in folder dist/html:

python -m sphinx -T -b html doc dist/html

Run the unit tests:

python -m unittest discover tests

Or:

python -m pytest

To check style:

python -m flake8 td3a_cpp_deep tests examples

The function check or the command line python -m td3a_cpp_deep check checks the module is properly installed and returns processing time for a couple of functions or simply:

import td3a_cpp_deep
td3a_cpp_deep.check()