triton源码分析之setup.py

youzjuer

已于 2024-06-04 16:40:59 修改

阅读量171

点赞数

分类专栏：通俗易懂技术站文章标签： python 开发语言

于 2024-05-29 13:51:55 首次发布

本文链接：https://blog.csdn.net/youzjuer/article/details/139293829

版权

通俗易懂技术站专栏收录该内容

71 篇文章 5 订阅 ¥59.90 ¥99.00

订阅专栏

超级会员免费看

本文详细分析了在执行`pip install -e .`时，triton源码中的`setup.py`文件执行流程。从cmdclass中的自定义命令到CMakeBuild过程，再到获取缓存路径和第三方包信息，特别是`get_pybind11_package_info`函数的功能，该函数负责提供pybind11的相关元信息，但实际下载发生在`get_thirdparty_packages`函数中。

摘要由CSDN通过智能技术生成

一执行流程

在执行pip install -e .的时候，便会执行这个文件，文件的入口为：

setup(
    name=os.environ.get("TRITON_WHEEL_NAME", "triton"),
    version="3.0.0" + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", ""),
    author="Philippe Tillet",
    author_email="phil@openai.com",
    description="A language and compiler for custom Deep Learning operations",
    long_description="",
    packages=get_packages(),
    entry_points=get_entry_points(),
    install_requires=get_install_requires(),
    package_data=package_data,
    include_package_data=True,
    ext_modules=[CMakeExtension("triton", "triton/_C/")]

了解本专栏