github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[extension] hotfix jit extension setup (#5402)

pull/5295/head
Hongxin Liu 2024-02-26 19:46:58 +08:00 committed by GitHub
parent 5d380a1a21
commit 95c21e3950
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
setup.py
View File

@ -121,8 +121,9 @@ setup(
"tests", "tests",
"scripts", "scripts",
"requirements", "requirements",
"extensions",
"*.egg-info", "*.egg-info",
) ),
), ),
description="An integrated large-scale model training system with efficient parallelization techniques", description="An integrated large-scale model training system with efficient parallelization techniques",
long_description=fetch_readme(), long_description=fetch_readme(),
@ -153,10 +154,7 @@ setup(
], ],
package_data={ package_data={
"colossalai": [ "colossalai": [
"_C/*.pyi", "kernel/extensions/csrc/**/*",
"kernel/cuda_native/csrc/*",
"kernel/cuda_native/csrc/kernel/*",
"kernel/cuda_native/csrc/kernels/include/*",
] ]
}, },
) )