state-spaces/extensions/kernels at main · tvanh512/state-spaces · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
benchmark_cauchy.py		benchmark_cauchy.py
benchmark_cauchy_tune.py		benchmark_cauchy_tune.py
cauchy.cpp		cauchy.cpp
cauchy.py		cauchy.py
cauchy_cuda.cu		cauchy_cuda.cu
map.h		map.h
setup.py		setup.py
test_cauchy.py		test_cauchy.py
test_vandermonde.py		test_vandermonde.py
tune_cauchy.py		tune_cauchy.py
tune_cauchy.sh		tune_cauchy.sh
tuner.py		tuner.py
tuning_setup.py		tuning_setup.py
vandermonde.py		vandermonde.py

README.md

Install the Cauchy and Vandermonde CUDA kernels:

python setup.py install

(Optional) Test the extensions

pytest -q -s test_cauchy.py
pytest -q -s test_vandermonde.py