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

[extension] hotfix compile check (#6099)

pull/6100/head
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
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():
compiled_before = True

2
extensions/cuda_extension.py
View File

@ -74,7 +74,7 @@ class _CudaExtension(_CppExtension):
# check if the kernel has been built
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():
compiled_before = True