github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 02:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

add colossalai kernel module (#55)

Browse Source
This commit is contained in:
shenggan
2021-12-21 12:19:52 +08:00
committed by GitHub
parent 648f806315
commit 5c3843dc98
43 changed files with 8329 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@@ -131,5 +131,6 @@ setup(
description='An integrated large-scale model training system with efficient parallelization techniques',
ext_modules=ext_modules,
cmdclass={'build_ext': BuildExtension} if ext_modules else {},
package_data={'colossalai': ['kernel/cuda_native/csrc/*']},
install_requires=install_requires,
)