()
| 488 | # as triton_dist._C.libtriton_distributed only when AOT is enabled. |
| 489 | ext_modules.append(CMakeExtension("triton_dist_csrc", os.path.join(get_base_dir(), "csrc"), "_C")) |
| 490 | |
| 491 | setup( |
| 492 | name=os.environ.get("TRITON_WHEEL_NAME", "triton_dist"), |
| 493 | # Aligned with the pinned upstream Triton (3.7.1) this plugin builds against. |
| 494 | version="3.7.1" + get_git_version_suffix() + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", ""), |
| 495 | author="ByteDance Seed", |
| 496 | author_email="zheng.size@bytedance.com", |
| 497 | description="Triton language and compiler extension for distributed deep learning systems " |
| 498 | "(out-of-tree Triton plugin).", |
| 499 | long_description="", |
| 500 | install_requires=["setuptools>=40.8.0", "packaging"] + TRITON_DEP + DEPS_NVIDIA + DEPS_HIP, |
| 501 | packages=get_packages(), |
| 502 | package_data=package_data, |
| 503 | include_package_data=True, |
| 504 | ext_modules=ext_modules, |
no test coverage detected