mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
[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:
@@ -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},
|
||||
|
||||
Reference in New Issue
Block a user