fix setup

6 months ago · 2d1a785b71
parent b96c6390f4
commit 2d1a785b71
1 changed files with 1 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@ -144,6 +144,7 @@ setup(
    package_data={
        "colossalai": [
            "kernel/extensions/csrc/**/*",
+            "kernel/extensions/pybind/**/*",
        ]
    },
 )