from setuptools import setup setup( name="mistralmlp_ext", packages=["mistralmlp_cuda"], package_data={"mistralmlp_cuda": ["*.so"]}, )