This is the legacy CUDA kernel for Power Attention, refer to https://github.com/m-a-n-i-f-e-s-t/power-attention instead.

Build

Build a single wheel

./build_wheels.sh -t <torch version> -p <python version>

Build all wheels

./build_all_wheels.sh

Build source dist (for pypi distribution)

python -m build --sdist