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