github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-25 14:55:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[devops] fix extention building (#5427)

Browse Source
This commit is contained in:
Hongxin Liu
2024-03-05 15:35:54 +08:00
committed by GitHub
parent 822241a99c
commit 070df689e6
24 changed files with 50 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -70,7 +70,7 @@ def get_version() -> str:
if BUILD_EXT:
if not TORCH_AVAILABLE:
raise ModuleNotFoundError(
"[extension] PyTorch is not found while CUDA_EXT=1. You need to install PyTorch first in order to build CUDA extensions"
"[extension] PyTorch is not found while BUILD_EXT=1. You need to install PyTorch first in order to build CUDA extensions"
)
from extensions import ALL_EXTENSIONS