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