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

[kernel] move all symlinks of kernel to colossalai._C (#1971)

Browse Source
This commit is contained in:
ver217
2022-11-17 13:42:33 +08:00
committed by GitHub
parent 7e24b9b9ee
commit f8a7148dec
27 changed files with 463 additions and 322 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/cli/check/check_installation.py
View File

@@ -1,5 +1,6 @@
import click
import subprocess
import click
import torch
from torch.utils.cpp_extension import CUDA_HOME
@@ -17,7 +18,7 @@ def check_installation():
def _check_cuda_extension_installed():
try:
import colossal_C
import colossalai._C.fused_optim
is_cuda_extension_installed = u'\u2713'
except ImportError:
is_cuda_extension_installed = 'x'