github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-04-28 03:43:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[extension] hotfix compile check (#6099)

Browse Source
This commit is contained in:
Hongxin Liu 2024-10-24 11:11:44 +08:00 committed by GitHub
parent dee63cc5ef
commit 80a8ca916a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/cpp_extension.py
View File

@ -79,7 +79,7 @@ class _CppExtension(_Extension):
# check if the kernel has been built # check if the kernel has been built
compiled_before = False compiled_before = False
kernel_file_path = build_directory.joinpath(f"{self.name}.o") kernel_file_path = build_directory.joinpath(f"{self.name}.so")
if kernel_file_path.exists(): if kernel_file_path.exists():
compiled_before = True compiled_before = True

2
extensions/cuda_extension.py
View File

@ -74,7 +74,7 @@ class _CudaExtension(_CppExtension):
# check if the kernel has been built # check if the kernel has been built
compiled_before = False compiled_before = False
kernel_file_path = build_directory.joinpath(f"{self.name}.o") kernel_file_path = build_directory.joinpath(f"{self.name}.so")
if kernel_file_path.exists(): if kernel_file_path.exists():
compiled_before = True compiled_before = True