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