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

[setup] refactored setup.py for dependency graph (#2413)

pull/2417/head
Frank Lee 2023-01-10 12:10:13 +08:00 committed by GitHub
parent 9d432230ba
commit 2445279a08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
setup.py
View File

@ -144,13 +144,16 @@ if build_cuda_ext:
print(f'===== Building Extension {name} =====') print(f'===== Building Extension {name} =====')
ext_modules.append(builder_cls().builder()) ext_modules.append(builder_cls().builder())
if is_nightly: # always put not nightly branch as the if branch
# otherwise github will treat colossalai-nightly as the project name
# and it will mess up with the dependency graph insights
if not is_nightly:
version = get_version()
package_name = 'colossalai'
else:
# use date as the nightly version # use date as the nightly version
version = datetime.today().strftime('%Y.%m.%d') version = datetime.today().strftime('%Y.%m.%d')
package_name = 'colossalai-nightly' package_name = 'colossalai-nightly'
else:
version = get_version()
package_name = 'colossalai'
setup(name=package_name, setup(name=package_name,
version=version, version=version,