[extension] hotfix compile check (#6099)

Author: ver217

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
 

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