setup.py

from setuptools import setup, find_packages

setup(
    name='MLAttention',
    packages=find_packages("triton_mla_op"),
    package_dir={"": "triton_mla_op"},
    include_package_data=True,
    install_requires=[
        'triton',
        'torch'
    ]
)