[extension] hotfix compile check (hpcaitech#6099)

TongLi3701 · Oct 24, 2024 · 80a8ca9 · 80a8ca9
1 parent dee63cc
commit 80a8ca9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/extensions/cpp_extension.py b/extensions/cpp_extension.py
@@ -79,7 +79,7 @@ def build_jit(self) -> None:
 
         # 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
 

diff --git a/extensions/cuda_extension.py b/extensions/cuda_extension.py
@@ -74,7 +74,7 @@ def build_jit(self) -> None:
 
         # 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