overlay/kernels/cuda/flashfftconv/setup.py

"""Python-wrapper setup for the vendored flashfftconv package.

This installs only the pure-Python wrappers in `flashfftconv/`. The actual
CUDA extension (`monarch_cuda`) must be built separately via `csrc/setup.py`
— see README.md.

License: Apache 2.0 (vendored from HazyResearch/flash-fft-conv).
"""

from setuptools import setup

if __name__ == "__main__":
    setup(
        name="flashfftconv",
        version="0.0.0+hydra-vendored",
        description="HazyResearch flash-fft-conv, vendored for HYDRA use",
        url="https://github.com/HazyResearch/flash-fft-conv",
        author="Dan Fu, Hermann Kumbong (upstream); vendored into HYDRA",
        license="Apache 2.0",
        packages=["flashfftconv"],
        package_dir={"flashfftconv": "flashfftconv"},
    )