fix setup

This commit is contained in:
flybird11111 2024-06-06 02:44:41 +00:00
parent b96c6390f4
commit 2d1a785b71

View File

@ -144,6 +144,7 @@ setup(
package_data={ package_data={
"colossalai": [ "colossalai": [
"kernel/extensions/csrc/**/*", "kernel/extensions/csrc/**/*",
"kernel/extensions/pybind/**/*",
] ]
}, },
) )