From fe15ce9f19d37e6c32bc05314c64acba33cd63b5 Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 24 Oct 2024 11:02:58 +0800 Subject: [PATCH] [extension] hotfix compile check --- extensions/cpp_extension.py | 2 +- extensions/cuda_extension.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions/cpp_extension.py b/extensions/cpp_extension.py index aaa43f964c25..a92195dda3d9 100644 --- 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 index da15bcd57a26..214b83ec83cc 100644 --- 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