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

[extension] hotfix compile check (#6099)

Browse Source
pull/6100/head
Hongxin Liu 1 month ago 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

2
extensions/cpp_extension.py
Unescape 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
Unescape 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

Write Preview
Loading…
Cancel
Save