from setuptools import setup

setup(
    name="mlp_ext",
    packages=["mlp_cuda"],
    package_data={"mlp_cuda": ["*.so"]},
)