[BUILD] Add missing subpackages to build (#1475)

The `triton/compiler`, `triton/runtime/driver`, and `triton/third_party`
subpackages were missing from the distribution built with the old
`setup.py` after #1464, causing an immediate error upon importing Triton
with a non-editable installation. This change adds the missing Python
subpackages and moves `triton/third_party` inclusion to `MANIFEST.in`,
where it will automatically be included in wheels due to the existing
`include_package_data` setup flag.
This commit is contained in:
Eta
2023-04-05 00:41:08 -05:00
committed by GitHub
parent 0e11f1e167
commit 577cafff0a
2 changed files with 7 additions and 4 deletions

View File

@@ -221,16 +221,18 @@ setup(
packages=[
"triton",
"triton/_C",
"triton/language",
"triton/tools",
"triton/common",
"triton/compiler",
"triton/language",
"triton/ops",
"triton/ops/blocksparse",
"triton/runtime",
"triton/ops/blocksparse"],
"triton/runtime/driver",
"triton/tools",
],
install_requires=[
"filelock",
],
package_data={"triton": ["third_party/**/*"]},
include_package_data=True,
ext_modules=[CMakeExtension("triton", "triton/_C/")],
cmdclass={"build_ext": CMakeBuild},