Skip to content

GitLab

Unverified Commit 95c21e39 authored by Hongxin Liu's avatar Hongxin Liu Committed by GitHub
Browse files

[extension] hotfix jit extension setup (#5402)

parent 5d380a1a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
setup.py
View file @ 95c21e39
...@@ -121,8 +121,9 @@ setup( ...@@ -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( ...@@ -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/*",
] ]
}, },
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment